Merge pull request #200 from yushan777/main

Update environment.yaml
JoePenna · Jan 8, 2024 · 4d4e744 · 4d4e744
2 parents 2d686f0 + 2bf9b44
commit 4d4e744
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