This repository was archived by the owner on Aug 21, 2025. It is now read-only.
File tree Expand file tree Collapse file tree 1 file changed +2
-8
lines changed
.circleci/unittest/linux/scripts Expand file tree Collapse file tree 1 file changed +2
-8
lines changed Original file line number Diff line number Diff line change @@ -55,15 +55,9 @@ pip install expecttest
55
55
if [ " ${CU_VERSION:- } " == cpu ] ; then
56
56
# conda install -y pytorch torchvision cpuonly -c pytorch-nightly
57
57
# use pip to install pytorch as conda can frequently pick older release
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
58
+ pip install torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --pre
62
59
PYTORCH_VERSION=" $( python -c " import torch; print(torch.__version__)" ) " python setup.py develop bdist_wheel -d $WHEELS_FOLDER
63
60
else
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
61
+ conda install -y pytorch== torchvision cudatoolkit=10.2 -c pytorch-nightly
68
62
PYTORCH_VERSION=" $( python -c " import torch; print(torch.__version__)" ) " python setup.py develop bdist_wheel -d $WHEELS_FOLDER
69
63
fi
You can’t perform that action at this time.
0 commit comments