Skip to content

Commit 8249db3

Browse files
authored
Fix CI by using pip instead of conda to install torch (#863)
1 parent 443ed83 commit 8249db3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,6 @@ if [ "${CU_VERSION:-}" == cpu ] ; then
5858
pip install torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --pre
5959
PYTORCH_VERSION="$(python -c "import torch; print(torch.__version__)")" python setup.py develop bdist_wheel -d $WHEELS_FOLDER
6060
else
61-
conda install -y pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly
61+
pip install torch torchvision -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html --pre
6262
PYTORCH_VERSION="$(python -c "import torch; print(torch.__version__)")" python setup.py develop bdist_wheel -d $WHEELS_FOLDER
6363
fi

0 commit comments

Comments
 (0)