summary refs log tree commit diff stats
path: root/classification/shell.nix
diff options
context:
space:
mode:
authorChristian Krinitsin <mail@krinitsin.com>2025-06-03 12:02:16 +0000
committerChristian Krinitsin <mail@krinitsin.com>2025-06-03 12:02:16 +0000
commite5317707f115fc8d6423e1d850c6b52d1f9edbb0 (patch)
treeb85dc15e9f82a0c189ff8b8a7fbbc5ee5c1350cc /classification/shell.nix
parent0d42327250d4b8d521ef1b0ef17ac673b2ae2ed4 (diff)
downloadqemu-analysis-e5317707f115fc8d6423e1d850c6b52d1f9edbb0.tar.gz
qemu-analysis-e5317707f115fc8d6423e1d850c6b52d1f9edbb0.zip
fix shell.nix
Diffstat (limited to 'classification/shell.nix')
-rw-r--r--classification/shell.nix3
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
   '';
 }