conda: fix pytorch packaging

agimus-project · Jan 12, 2024 · 4b0c521 · 4b0c521
1 parent 74faf2d
commit 4b0c521
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/environment.yml b/environment.yml
@@ -6,10 +6,10 @@ channels:
    - anaconda
    - defaults
 dependencies:
-  - nvidia::cudatoolkit==12.1
+  - pytorch-cuda==12.1
   - python=3.9
   - pip
-  - pytorch::pytorch==2.1
+  - pytorch==2.1
   - torchvision==0.16
   - geckodriver
   - firefox