Add torchvision to nix develop environment

EMI-Group · Jan 8, 2025 · b06cb5a · b06cb5a
1 parent 83aa4b6
commit b06cb5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dev_env_builder.nix b/dev_env_builder.nix
@@ -25,6 +25,7 @@ pkgs.mkShell rec {
     # add them to PYTHONPATH and thus make them accessible from within the venv.
     numpy
     torch
+    torchvision
   ] ++ (with pkgs; [
     pre-commit
     ruff
@@ -44,4 +45,3 @@ pkgs.mkShell rec {
   '';
 
 }
-