File tree Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -3,19 +3,25 @@ sudo apt-get install -y --no-install-recommends unzip p7zip-full sox libsox-dev
3
3
4
4
# Install a nightly build of pytorch
5
5
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
8
14
9
15
# 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
11
17
12
18
# # Install doc dependencies
13
19
14
20
export PATH=/opt/conda/bin:$PATH
15
21
# pillow >= 4.2 will throw error when trying to write mode RGBA as JPEG,
16
22
# this is a workaround to the issue.
17
23
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
19
25
20
26
git clone https://github.com/pytorch/vision --quiet
21
27
pushd vision
You can’t perform that action at this time.
0 commit comments