Revert "Upgrade pytorch/torchio (#313)"

This reverts commit 67c08db.
eigenvivek · Aug 1, 2024 · 0702325 · 0702325
1 parent 67c08db
commit 0702325
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/environment.yml b/environment.yml
@@ -5,12 +5,12 @@ channels:
   - nvidia
 dependencies:
   - pip
-  - pytorch
+  - pytorch>2.2,<2.3
   - einops
   - torchvision
   - torchio
   - timm
-  - numpy
+  - numpy<2
   - matplotlib
   - seaborn
   - imageio

diff --git a/settings.ini b/settings.ini
@@ -26,9 +26,9 @@ keywords = nbdev jupyter notebook python
 language = English
 status = 3
 user = eigenvivek
-requirements = matplotlib seaborn tqdm imageio fastcore 'pyvista[all]' einops torchvision scipy torchio timm numpy
-pip_requirements = torch
-conda_requirements = pytorch
+requirements = matplotlib seaborn tqdm imageio fastcore 'pyvista[all]' einops torchvision==0.17.2 scipy torchio timm numpy==1.26.4
+pip_requirements = torch==2.2.2
+conda_requirements = pytorch==2.2.2
 dev_requirements = nbdev black flake8 ipykernel ipywidgets jupyterlab jupyterlab_execute_time jupyterlab-code-formatter isort
 readme_nb = index.ipynb
 allowed_metadata_keys =