Updated environment files because pytorch repos changed

AnnikaStein · Nov 9, 2023 · 8bd33bd · 8bd33bd
1 parent 3d65e92
commit 8bd33bd
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 8 deletions.
diff --git a/environment_cuda117.yaml b/environment_cuda117.yaml
@@ -2,15 +2,14 @@ name: spanet
 
 channels:
   - pytorch
-  - nvidia/label/cuda-11.7.0
+  - nvidia
   - conda-forge
   - nodefaults
 
 dependencies:
   - python=3.10.*
   - conda
   - pip
-  - cuda-toolkit
   - onnx
 
   - numpy=1.24

diff --git a/environment_cuda118.yaml b/environment_cuda118.yaml
@@ -1,16 +1,15 @@
 name: spanet
 
 channels:
-  - pytorch-nightly
-  - nvidia/label/cuda-11.8.0
+  - pytorch
+  - nvidia
   - conda-forge
   - nodefaults
 
 dependencies:
   - python=3.10.*
   - conda
   - pip
-  - cuda-toolkit
   - onnx
 
   - numpy=1.24

diff --git a/environment_cuda121.yaml b/environment_cuda121.yaml
@@ -1,16 +1,15 @@
 name: spanet
 
 channels:
-  - pytorch-nightly
-  - nvidia/label/cuda-12.1.0
+  - pytorch
+  - nvidia
   - conda-forge
   - nodefaults
 
 dependencies:
   - python=3.10.*
   - conda
   - pip
-  - cuda-toolkit
   - onnx
 
   - numpy=1.24