Adds an extra index url for cpu downloading

MachineLearningLifeScience · Aug 5, 2024 · c5f0918 · c5f0918
1 parent bd572de
commit c5f0918
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/poli_baselines/solvers/bayesian_optimization/pr/environment.pr.small.yml b/src/poli_baselines/solvers/bayesian_optimization/pr/environment.pr.small.yml
@@ -6,8 +6,8 @@ dependencies:
   - pytorch
   - pip
   - pip:
+      - --extra-index-url https://download.pytorch.org/whl/cpu
       - numpy<2
-      - cpuonly
       - torch
       - gpytorch==1.11
       - botorch==0.7.0