blob: 5a20b8e463572e0268e9b755312403bc075161f4 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
{ pkgs ? (import <nixpkgs> {}).pkgs }:
with pkgs;
mkShell {
buildInputs = [
python312Packages.transformers
python312Packages.torch
];
shellHook = ''
# fixes libstdc++ issues and libgl.so issues
LD_LIBRARY_PATH=${stdenv.cc.cc.lib}/lib/:/run/opengl-driver/lib/
# fixes unallocation issues
export MKL_NUM_THREADS=64
'';
}
|