Update packaging to torch 2 / CUDA 12.1

agimus-project · Jan 12, 2024 · 74faf2d · 74faf2d
1 parent f5f1faf
commit 74faf2d
Show file tree

Hide file tree

Showing 3 changed files with 659 additions and 624 deletions.
diff --git a/environment.yml b/environment.yml
@@ -6,11 +6,11 @@ channels:
    - anaconda
    - defaults
 dependencies:
-  - nvidia::cudatoolkit==11.3.1
+  - nvidia::cudatoolkit==12.1
   - python=3.9
   - pip
-  - pytorch::pytorch==1.11.0
-  - torchvision==0.12.0
+  - pytorch::pytorch==2.1
+  - torchvision==0.16
   - geckodriver
   - firefox
   - opencv