try to fix

Author: Will Feng

.circleci/config.yml

@@ -162,14 +162,14 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults
 
 pytorch_tutorial_build_worker_defaults: &pytorch_tutorial_build_worker_defaults
   environment:
-    DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda9-cudnn7-py3"
-    CUDA_VERSION: "9"
+    DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10-cudnn7-py3-gcc7"
+    CUDA_VERSION: "10"
   resource_class: gpu.medium
   <<: *pytorch_tutorial_build_defaults
 
 pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_defaults
   environment:
-    DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda9-cudnn7-py3"
+    DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda10-cudnn7-py3-gcc7"
   resource_class: medium
   <<: *pytorch_tutorial_build_defaults
 

.jenkins/build.sh

@@ -25,12 +25,13 @@ pip install -e git+git://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch
 # this is a workaround to the issue.
 pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1
 
-# Install torchvision from source
-git clone https://github.com/pytorch/vision --quiet
-pushd vision
-git checkout 76b2667958f994b7d942645285f777046d7d2d16
-pip install . --no-deps  # We don't want it to install the stock PyTorch version from pip
-popd
+# Install torchvision from conda (and install torch along with it)
+if [[ $(pip show torch) ]]; then
+  # Clean up previous PyTorch installations
+  pip uninstall -y torch || true
+  pip uninstall -y torch || true
+fi
+conda install -c pytorch torchvision
 
 # Install torchaudio from source
 git clone https://github.com/pytorch/audio --quiet
@@ -40,17 +41,6 @@ popd
 
 aws configure set default.s3.multipart_threshold 5120MB
 
-if [[ $(pip show torch) ]]; then
-  # Clean up previous PyTorch installations
-  pip uninstall -y torch || true
-  pip uninstall -y torch || true
-fi
-
-# Install a nightly build of pytorch
-
-# GPU, requires CUDA version 9.0
-pip install cython torch_nightly -f https://download.pytorch.org/whl/nightly/cu90/torch_nightly.html
-
 # Decide whether to parallelize tutorial builds, based on $JOB_BASE_NAME
 export NUM_WORKERS=20
 if [[ "${JOB_BASE_NAME}" == *worker_* ]]; then