Add torch and torchvision to environment.yml

c-salomonsen · c-salomonsen · commit 85658e7e26ac · 2025-02-01T18:34:20.000+01:00
diff --git a/environment.yml b/environment.yml
@@ -18,5 +18,8 @@ dependencies:
   - pytest
   - ruff
   - scalene
+  - pip:
+    - torch
+    - torchvision
 prefix: /opt/miniconda3/envs/cc-exam
 
