From 4982ee71932a23f38a47f940d9e64d3efee3c77a Mon Sep 17 00:00:00 2001 From: thorstenwagner Date: Fri, 13 Oct 2023 15:29:13 +0200 Subject: [PATCH 1/2] use stable pytorch 2.1 --- conda_env_tomotwin.yml | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/conda_env_tomotwin.yml b/conda_env_tomotwin.yml index cc7a976..6ba866d 100644 --- a/conda_env_tomotwin.yml +++ b/conda_env_tomotwin.yml @@ -6,6 +6,8 @@ channels: - conda-forge - defaults dependencies: + - pytorch[version='>=2.1'] + - torchvision - pandas[version='<2'] - scipy - numpy @@ -19,8 +21,5 @@ dependencies: - mysql-connector-python - pip - pip: - - --extra-index-url https://download.pytorch.org/whl/nightly/cu118 - - torch==2.1.0.dev20230822+cu118 - - torchvision==0.16.0.dev20230822 - tomotwin-cryoet From 270f6fef43caef0f536a83cc6848f2a6dfac7c00 Mon Sep 17 00:00:00 2001 From: thorstenwagner Date: Fri, 13 Oct 2023 15:33:47 +0200 Subject: [PATCH 2/2] use stable pytorch channel --- conda_env_tomotwin.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/conda_env_tomotwin.yml b/conda_env_tomotwin.yml index 6ba866d..abfb72d 100644 --- a/conda_env_tomotwin.yml +++ b/conda_env_tomotwin.yml @@ -1,7 +1,7 @@ name: tomotwin channels: - nvidia - - pytorch-nightly + - pytorch - rapidsai - conda-forge - defaults