Skip to content

Commit 7ddac90

Browse files
committed
remove_pinned
1 parent 70e8b1b commit 7ddac90

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

.github/scripts/setup-env.sh

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,6 @@ conda activate ci
3636
conda install --quiet --yes libjpeg-turbo -c pytorch
3737
pip install --progress-bar=off --upgrade setuptools==72.1.0
3838

39-
4039
# See https://github.com/pytorch/vision/issues/6790
4140
if [[ "${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
;;
7978
esac
8079
PYTORCH_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

8682
if [[ $GPU_ARCH_TYPE == 'cuda' ]]; then
8783
python -c "import torch; exit(not torch.cuda.is_available())"

0 commit comments

Comments
 (0)