CI: install pytorch from conda-forge

mj-will · Jan 23, 2025 · 0303500 · 0303500
1 parent d45320f
commit 0303500
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/ci_environment.yaml b/ci_environment.yaml
@@ -1,15 +1,13 @@
 name: nessai-ci
 channels:
-  - pytorch
   - conda-forge
 dependencies:
   - numpy
   - pandas
   - scipy>=0.16
   - matplotlib>=2.0
   - seaborn
-  - pytorch>=1.11.0
-  - cpuonly
+  - pytorch-cpu>=2.0
   - glasflow
   - h5py
   - corner