Skip to content

Commit be5b9ce

Browse files
committed
Update the PT1 Tutorials Jenkins Build script to use CUDA 8.0
1 parent a61996a commit be5b9ce

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

.jenkins/build.sh

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,17 @@ sudo apt-get install -y --no-install-recommends unzip p7zip-full sox libsox-dev
33

44
# Install a nightly build of pytorch
55

6-
# GPU, requires CUDA version 9.0 and python version 3.6
7-
pip install cython torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly-2018.8.28.dev1-cp36-cp36m-linux_x86_64.whl
6+
# GPU, requires CUDA version 8.0
7+
pip install cython torch_nightly -f pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu80/torch_nightly.html
8+
9+
# GPU, requires CUDA version 9.0
10+
# pip install cython torch_nightly -f pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html
11+
12+
# GPU, requires CUDA version 9.2
13+
# pip install cython torch_nightly -f pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu92/torch_nightly.html
814

915
# CPU
10-
# pip install cython torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly-2018.8.24.dev1-cp36-cp36m-linux_x86_64.whl
16+
# pip install cython torch_nightly -f pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
1117

1218
## Install doc dependencies
1319

0 commit comments

Comments
 (0)