prepare-env.sh: force installing cpu version of pytorch

pengyunie · Jan 19, 2023 · 42e2233 · 42e2233
1 parent 503f7b9
commit 42e2233
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/prepare-env.sh b/prepare-env.sh
@@ -34,6 +34,7 @@ function prepare_conda_env() {
         conda activate $env_name
 
         # Install libraries
+        pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
         pip install -e .[dev,io-3rd-party]
 }