Skip to content

Commit c9ba905

Browse files
author
Will Feng
committed
fix
1 parent df9a4e4 commit c9ba905

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.jenkins/build.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,8 @@ pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1
2727
# Install torchvision from conda
2828
pip uninstall -y torchvision || true
2929
pip uninstall -y torchvision || true
30-
conda install -c pytorch torchvision --no-deps
30+
# GPU, requires CUDA version 9.0
31+
conda install https://anaconda.org/pytorch/torchvision/0.3.0/download/linux-64/torchvision-0.3.0-py36_cu9.0.176_1.tar.bz2 --no-deps
3132

3233
# Install torchaudio from source
3334
git clone https://github.com/pytorch/audio --quiet
@@ -44,7 +45,6 @@ if [[ $(pip show torch) ]]; then
4445
fi
4546

4647
# Install a nightly build of pytorch
47-
4848
# GPU, requires CUDA version 9.0
4949
pip install cython torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html
5050

0 commit comments

Comments
 (0)