Merge branch 'main' into tar-generator

NikoOinonen · NikoOinonen · commit 8706789e9f05 · 2025-02-06T16:13:41.000+02:00
diff --git a/environment.yml b/environment.yml
@@ -1,15 +1,14 @@
 name: ml-spm
 channels:
   - pytorch
-  - nvidia/label/cuda-11.8.0
+  - nvidia/label/cuda-12.1.0
   - conda-forge
   - defaults
 dependencies:
   - python=3.11
   - pip
   - pytorch
-  - torchvision
-  - torchaudio
+  - pytorch-cuda=12.1
   - cuda
   - ninja
   - matplotlib
