change torch version

isl-org · Aug 20, 2024 · aa4afe9 · aa4afe9
1 parent 023a166
commit aa4afe9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker_entrypoint.sh b/docker_entrypoint.sh
@@ -12,7 +12,7 @@ conda create -y -n asr python=$PYTHON_VERSION
 source activate asr
 conda install -y cmake
 
-python -m pip install torch==1.8.2+cpu -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
+python -m pip install torch==1.13.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu
 python -m pip install open3d==0.14.1 zstandard msgpack msgpack-numpy
 
 pushd datasets