From ef324e0fa7acdfa42a5ac2ba817c1f217fbc8512 Mon Sep 17 00:00:00 2001 From: Benjamin Kiessling Date: Fri, 9 Feb 2024 16:09:46 +0100 Subject: [PATCH] incorrect pytorch pin in conda envs --- environment.yml | 2 +- environment_cuda.yml | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/environment.yml b/environment.yml index fe94bd329..9b4baa3df 100644 --- 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 index fe364ed43..826d49d76 100644 --- 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