diff options
Diffstat (limited to 'classification/shell.nix')
| -rw-r--r-- | classification/shell.nix | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/classification/shell.nix b/classification/shell.nix index bab1e7d2a..5a20b8e46 100644 --- a/classification/shell.nix +++ b/classification/shell.nix @@ -2,12 +2,13 @@ with pkgs; mkShell { buildInputs = [ - python312Packages.numpy 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 ''; } |