File tree Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Original file line number Diff line number Diff line change @@ -36,7 +36,6 @@ conda activate ci
3636conda install --quiet --yes libjpeg-turbo -c pytorch
3737pip install --progress-bar=off --upgrade setuptools==72.1.0
3838
39-
4039# See https://github.com/pytorch/vision/issues/6790
4140if [[ " ${PYTHON_VERSION} " != " 3.11" ]]; then
4241 pip install --progress-bar=off av! =10.0.0
@@ -78,10 +77,7 @@ case $GPU_ARCH_TYPE in
7877 ;;
7978esac
8079PYTORCH_WHEEL_INDEX=" https://download.pytorch.org/whl/${CHANNEL} /${GPU_ARCH_ID} "
81- # pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}"
82-
83- # install torch built with manylinux 2.28
84- pip install --progress-bar=off --pre torch==2.6.0.dev20241106 --index-url=" ${PYTORCH_WHEEL_INDEX} " --force-reinstall
80+ pip install --progress-bar=off --pre torch --index-url=" ${PYTORCH_WHEEL_INDEX} "
8581
8682if [[ $GPU_ARCH_TYPE == ' cuda' ]]; then
8783 python -c " import torch; exit(not torch.cuda.is_available())"
You can’t perform that action at this time.
0 commit comments