@@ -25,12 +25,13 @@ pip install -e git+git://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch
25
25
# this is a workaround to the issue.
26
26
pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1
27
27
28
- # Install torchvision from source
29
- git clone https://github.com/pytorch/vision --quiet
30
- pushd vision
31
- git checkout 76b2667958f994b7d942645285f777046d7d2d16
32
- pip install . --no-deps # We don't want it to install the stock PyTorch version from pip
33
- popd
28
+ # Install torchvision from conda (and install torch along with it)
29
+ if [[ $( pip show torch) ]]; then
30
+ # Clean up previous PyTorch installations
31
+ pip uninstall -y torch || true
32
+ pip uninstall -y torch || true
33
+ fi
34
+ conda install -c pytorch torchvision
34
35
35
36
# Install torchaudio from source
36
37
git clone https://github.com/pytorch/audio --quiet
40
41
41
42
aws configure set default.s3.multipart_threshold 5120MB
42
43
43
- if [[ $( pip show torch) ]]; then
44
- # Clean up previous PyTorch installations
45
- pip uninstall -y torch || true
46
- pip uninstall -y torch || true
47
- fi
48
-
49
- # Install a nightly build of pytorch
50
-
51
- # GPU, requires CUDA version 9.0
52
- pip install cython torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html
53
-
54
44
# Decide whether to parallelize tutorial builds, based on $JOB_BASE_NAME
55
45
export NUM_WORKERS=20
56
46
if [[ " ${JOB_BASE_NAME} " == * worker_* ]]; then
0 commit comments