Skip to content

Commit

Permalink
remove_pinned
Browse files Browse the repository at this point in the history
  • Loading branch information
atalman committed Nov 14, 2024
1 parent 70e8b1b commit 7ddac90
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions .github/scripts/setup-env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,6 @@ conda activate ci
conda install --quiet --yes libjpeg-turbo -c pytorch
pip install --progress-bar=off --upgrade setuptools==72.1.0


# See https://github.com/pytorch/vision/issues/6790
if [[ "${PYTHON_VERSION}" != "3.11" ]]; then
pip install --progress-bar=off av!=10.0.0
Expand Down Expand Up @@ -78,10 +77,7 @@ case $GPU_ARCH_TYPE in
;;
esac
PYTORCH_WHEEL_INDEX="https://download.pytorch.org/whl/${CHANNEL}/${GPU_ARCH_ID}"
#pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}"

#install torch built with manylinux 2.28
pip install --progress-bar=off --pre torch==2.6.0.dev20241106 --index-url="${PYTORCH_WHEEL_INDEX}" --force-reinstall
pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}"

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

0 comments on commit 7ddac90

Please sign in to comment.