diff --git a/ci/scripts/ubuntu_build_test.sh b/ci/scripts/ubuntu_build_test.sh index b68b2218be2c7..4a2c42b6ca3a0 100755 --- a/ci/scripts/ubuntu_build_test.sh +++ b/ci/scripts/ubuntu_build_test.sh @@ -23,7 +23,7 @@ cd taichi git checkout $SHA python3 -m pip install -r requirements_dev.txt -i http://repo.taichigraphics.com/repository/pypi/simple --trusted-host repo.taichigraphics.com # Update Torch version, otherwise cuda tests fail. See #2969. -python3 -m pip install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html -i http://repo.taichigraphics.com/repository/pypi/simple --trusted-host repo.taichigraphics.com +python3 -m pip install torch==1.10.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html -i http://repo.taichigraphics.com/repository/pypi/simple --trusted-host repo.taichigraphics.com TAICHI_CMAKE_ARGS="-DTI_WITH_VULKAN:BOOL=ON -DTI_WITH_CUDA:BOOL=ON -DTI_WITH_OPENGL:BOOL=ON" python3 setup.py install # Add Docker specific ENV