Update environment.yml

eigenvivek · Jun 28, 2024 · 4ed8df3 · 4ed8df3
1 parent ec7487f
commit 4ed8df3
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/environment.yml b/environment.yml
@@ -6,16 +6,15 @@ channels:
 dependencies:
   - pip
   - pytorch
-  - torchvision
+  - torchvision<0.18
+  - torchio>=0.19.8
   - einops
+  - timm
   - matplotlib
   - seaborn
   - imageio
   - fastcore
-  - tqdm
+  - pyvista
+  - vtk
   - scipy
-  - pip:
-    - pyvista
-    - timm
-    - torchio>=0.19.8
-    - vtk
+  - tqdm