incorrect pytorch pin in conda envs

mittagessen · Feb 9, 2024 · ef324e0 · ef324e0
1 parent db286ec
commit ef324e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/environment.yml b/environment.yml
@@ -14,7 +14,7 @@ dependencies:
   - scipy~=1.10.0
   - jinja2~=3.0
   - conda-forge::torchvision-cpu>=0.5.0
-  - conda-forge::pytorch-cpu>=1.11,<1.14
+  - conda-forge::pytorch-cpu~=2.0.1
   - jsonschema
   - scikit-learn~=1.2.1
   - scikit-image~=0.21.0

diff --git a/environment_cuda.yml b/environment_cuda.yml
@@ -13,8 +13,8 @@ dependencies:
   - pillow>=9.2.0
   - scipy~=1.10.0
   - jinja2~=3.0
-  - conda-forge::torchvision
-  - conda-forge::pytorch>=1.11,<1.14
+  - conda-forge::torchvision>=0.5.0
+  - conda-forge::pytorch~=2.0.1
   - cudatoolkit>=9.2
   - jsonschema
   - scikit-learn~=1.2.1