Skip to content

Commit a681aeb

Browse files
author
Will Feng
authored
Fix torchvision install (pytorch#508)
* Fix torchvision install * update * try to fix * try to fix * fix * use cuda 10 * try to fix * Revert "try to fix" This reverts commit 5a610de. * Revert "use cuda 10" This reverts commit 013548a. * Revert "fix" This reverts commit c9ba905. * Revert "try to fix" This reverts commit df9a4e4. * Revert "try to fix" This reverts commit 49c65a7. * Revert "update" This reverts commit ac0cc5b. * Revert "Fix torchvision install" This reverts commit 2eca83c. * try another way * try to fix * try to fix * try something different
1 parent 400e324 commit a681aeb

File tree

2 files changed

+1
-18
lines changed

2 files changed

+1
-18
lines changed

.jenkins/build.sh

-18
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@ sudo apt-get install -y --no-install-recommends unzip p7zip-full sox libsox-dev
1414
export PATH=/opt/conda/bin:$PATH
1515
rm -rf src
1616
pip install -r $DIR/../requirements.txt
17-
pip uninstall -y torchvision || true
1817

1918
export PATH=/opt/conda/bin:$PATH
2019
conda install -y sphinx==1.8.2 pandas
@@ -25,12 +24,6 @@ pip install -e git+git://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch
2524
# this is a workaround to the issue.
2625
pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1
2726

28-
# Install torchvision from source
29-
git clone https://github.com/pytorch/vision --quiet
30-
pushd vision
31-
pip install . --no-deps # We don't want it to install the stock PyTorch version from pip
32-
popd
33-
3427
# Install torchaudio from source
3528
git clone https://github.com/pytorch/audio --quiet
3629
pushd audio
@@ -39,17 +32,6 @@ popd
3932

4033
aws configure set default.s3.multipart_threshold 5120MB
4134

42-
if [[ $(pip show torch) ]]; then
43-
# Clean up previous PyTorch installations
44-
pip uninstall -y torch || true
45-
pip uninstall -y torch || true
46-
fi
47-
48-
# Install a nightly build of pytorch
49-
50-
# GPU, requires CUDA version 9.0
51-
pip install cython torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html
52-
5335
# Decide whether to parallelize tutorial builds, based on $JOB_BASE_NAME
5436
export NUM_WORKERS=20
5537
if [[ "${JOB_BASE_NAME}" == *worker_* ]]; then

requirements.txt

+1
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ sphinx-gallery
55
tqdm
66
numpy
77
matplotlib
8+
torch
89
torchvision
910
PyHamcrest
1011
bs4

0 commit comments

Comments
 (0)