Skip to content

Commit f942191

Browse files
authored
Merge pull request #308 from ericnakagawa/rc1
Update the PT1 Tutorials Jenkins Build script to use CUDA 8.0
2 parents a61996a + ae4074d commit f942191

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

.jenkins/build.sh

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,19 +3,25 @@ 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

1420
export PATH=/opt/conda/bin:$PATH
1521
# pillow >= 4.2 will throw error when trying to write mode RGBA as JPEG,
1622
# this is a workaround to the issue.
1723
conda install -y sphinx pandas pillow=4.1.1
18-
pip install sphinx-gallery sphinx_rtd_theme tqdm matplotlib ipython
24+
pip install sphinx-gallery pytorch_sphinx_theme tqdm matplotlib ipython
1925

2026
git clone https://github.com/pytorch/vision --quiet
2127
pushd vision

0 commit comments

Comments
 (0)