Update environment.yaml

Fixes 'Torch not compiled with CUDA' error
JoePenna · Jan 8, 2024 · 2bf9b44 · 2bf9b44
1 parent 2d686f0
commit 2bf9b44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/environment.yaml b/environment.yaml
@@ -8,9 +8,9 @@ dependencies:
   - python=3.10
   - pip=22.2.2
   - cudatoolkit=11.8
-  - pytorch
+  - pytorch=2.0.1
   - torchvision
-  - pytorch-cuda=11.7
+  - pytorch-cuda=11.8
   - numpy=1.23.1
   - pip:
     - pytorch-lightning==1.7.6