Fix pytorch version to 2.0.1 in conda yaml file

HPCForge · Jan 31, 2024 · 389e711 · 389e711
1 parent 25abfbf
commit 389e711
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/conda/pytorch-2.0.1-cuda-11.7.yaml b/conda/pytorch-2.0.1-cuda-11.7.yaml
@@ -6,7 +6,7 @@ channels:
   - pyg
 dependencies:
   - python=3.9
-  - pytorch
+  - pytorch=2.0.1
   - torchvision
   - pytorch-cuda=11.7
   - pyg