Skip to content
This repository was archived by the owner on Aug 21, 2025. It is now read-only.

Commit 0057b5f

Browse files
authored
temporary fix pth version to make CI pass (#174)
* temporary fix pth version to make CI pass * Update install.sh
1 parent d451ee5 commit 0057b5f

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

.circleci/unittest/linux/scripts/install.sh

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,15 @@ pip install expecttest
5555
if [ "${CU_VERSION:-}" == cpu ] ; then
5656
# conda install -y pytorch torchvision cpuonly -c pytorch-nightly
5757
# use pip to install pytorch as conda can frequently pick older release
58-
pip install torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --pre
58+
# TEMPORARY FIX PYTORCH VERSION DUE TO REVERTED log_softmax PR
59+
# https://github.com/facebookresearch/functorch/pull/160#issuecomment-932389442
60+
# pip install torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --pre
61+
pip install torch==1.11.0.dev20211001+cpu torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --pre
5962
PYTORCH_VERSION="$(python -c "import torch; print(torch.__version__)")" python setup.py develop bdist_wheel -d $WHEELS_FOLDER
6063
else
61-
conda install -y pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly
64+
# conda install -y pytorch== torchvision cudatoolkit=10.2 -c pytorch-nightly
65+
# TEMPORARY FIX PYTORCH VERSION DUE TO REVERTED log_softmax PR
66+
# https://github.com/facebookresearch/functorch/pull/160#issuecomment-932389442
67+
pip install torch==1.11.0.dev20211001+cu102 torchvision -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html --pre
6268
PYTORCH_VERSION="$(python -c "import torch; print(torch.__version__)")" python setup.py develop bdist_wheel -d $WHEELS_FOLDER
6369
fi

0 commit comments

Comments
 (0)