From 2eca83c623a1441cb3578ecc44adc929edbff800 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 12:29:05 -0400 Subject: [PATCH 01/18] Fix torchvision install --- .jenkins/build.sh | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index dc8431c9dda..73217bdf07c 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -25,11 +25,8 @@ 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 -pip install . --no-deps # We don't want it to install the stock PyTorch version from pip -popd +# Install torchvision from pip +pip install torchvision --no-deps # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet From ac0cc5b81aadae6eaefb3cdf8d7a16fca29b7ecf Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 12:51:32 -0400 Subject: [PATCH 02/18] update --- .circleci/config.yml | 2 +- .jenkins/build.sh | 7 ++++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index a84b1eef2e8..7125547ca60 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -113,7 +113,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults command: | set -e - export pyTorchDockerImageTag=291 + export pyTorchDockerImageTag=300 echo "PyTorchDockerImageTag: "${pyTorchDockerImageTag} cat >/home/circleci/project/ci_build_script.sh < Date: Thu, 23 May 2019 13:47:04 -0400 Subject: [PATCH 03/18] try to fix --- .jenkins/build.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 0323ad20f99..3a82a663e31 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -27,7 +27,7 @@ pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1 # Install torchvision from conda pip uninstall -y torchvision || true pip uninstall -y torchvision || true -conda install -c pytorch torchvision +conda install -c pytorch torchvision --no-deps # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet From df9a4e4b77d092e0a189b85d315e26edc5ac025f Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 16:16:41 -0400 Subject: [PATCH 04/18] try to fix --- .circleci/config.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 7125547ca60..a84b1eef2e8 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -113,7 +113,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults command: | set -e - export pyTorchDockerImageTag=300 + export pyTorchDockerImageTag=291 echo "PyTorchDockerImageTag: "${pyTorchDockerImageTag} cat >/home/circleci/project/ci_build_script.sh < Date: Thu, 23 May 2019 17:22:14 -0400 Subject: [PATCH 05/18] fix --- .jenkins/build.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 3a82a663e31..5b55ffd980b 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -27,7 +27,8 @@ pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1 # Install torchvision from conda pip uninstall -y torchvision || true pip uninstall -y torchvision || true -conda install -c pytorch torchvision --no-deps +# GPU, requires CUDA version 9.0 +conda install https://anaconda.org/pytorch/torchvision/0.3.0/download/linux-64/torchvision-0.3.0-py36_cu9.0.176_1.tar.bz2 --no-deps # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet @@ -44,7 +45,6 @@ if [[ $(pip show torch) ]]; then 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 From 013548a8278c94634f71902caeac9617bef86369 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 17:39:03 -0400 Subject: [PATCH 06/18] use cuda 10 --- .circleci/config.yml | 6 +++--- .jenkins/build.sh | 8 ++++---- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index a84b1eef2e8..cb0e717a445 100644 --- a/.circleci/config.yml +++ b/.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 diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 5b55ffd980b..71660600b14 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -27,8 +27,8 @@ pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1 # Install torchvision from conda pip uninstall -y torchvision || true pip uninstall -y torchvision || true -# GPU, requires CUDA version 9.0 -conda install https://anaconda.org/pytorch/torchvision/0.3.0/download/linux-64/torchvision-0.3.0-py36_cu9.0.176_1.tar.bz2 --no-deps +# GPU, requires CUDA version 10.0 +conda install -c pytorch torchvision --no-deps # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet @@ -45,8 +45,8 @@ if [[ $(pip show torch) ]]; then 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 +# GPU, requires CUDA version 10.0 +pip install cython torch_nightly -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html # Decide whether to parallelize tutorial builds, based on $JOB_BASE_NAME export NUM_WORKERS=20 From 5a610de9c015824454a3da1ea71b5fd5d2547c53 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 18:02:33 -0400 Subject: [PATCH 07/18] try to fix --- .jenkins/build.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 71660600b14..a00bd9bdfd9 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -28,7 +28,7 @@ pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1 pip uninstall -y torchvision || true pip uninstall -y torchvision || true # GPU, requires CUDA version 10.0 -conda install -c pytorch torchvision --no-deps +conda install -y -c pytorch torchvision --no-deps # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet From 6753128b1d4681c379d8eace11b337b3d5fbcf88 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 18:26:02 -0400 Subject: [PATCH 08/18] Revert "try to fix" This reverts commit 5a610de9c015824454a3da1ea71b5fd5d2547c53. --- .jenkins/build.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index a00bd9bdfd9..71660600b14 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -28,7 +28,7 @@ pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1 pip uninstall -y torchvision || true pip uninstall -y torchvision || true # GPU, requires CUDA version 10.0 -conda install -y -c pytorch torchvision --no-deps +conda install -c pytorch torchvision --no-deps # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet From 9dee719413526d30e0a685f641f2616047f09cfc Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 18:26:09 -0400 Subject: [PATCH 09/18] Revert "use cuda 10" This reverts commit 013548a8278c94634f71902caeac9617bef86369. --- .circleci/config.yml | 6 +++--- .jenkins/build.sh | 8 ++++---- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index cb0e717a445..a84b1eef2e8 100644 --- a/.circleci/config.yml +++ b/.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-cuda10-cudnn7-py3-gcc7" - CUDA_VERSION: "10" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda9-cudnn7-py3" + CUDA_VERSION: "9" 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-cuda10-cudnn7-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda9-cudnn7-py3" resource_class: medium <<: *pytorch_tutorial_build_defaults diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 71660600b14..5b55ffd980b 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -27,8 +27,8 @@ pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1 # Install torchvision from conda pip uninstall -y torchvision || true pip uninstall -y torchvision || true -# GPU, requires CUDA version 10.0 -conda install -c pytorch torchvision --no-deps +# GPU, requires CUDA version 9.0 +conda install https://anaconda.org/pytorch/torchvision/0.3.0/download/linux-64/torchvision-0.3.0-py36_cu9.0.176_1.tar.bz2 --no-deps # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet @@ -45,8 +45,8 @@ if [[ $(pip show torch) ]]; then fi # Install a nightly build of pytorch -# GPU, requires CUDA version 10.0 -pip install cython torch_nightly -f https://download.pytorch.org/whl/nightly/cu100/torch_nightly.html +# 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 From ecd7683eb5f7f3091b91df7504c81d78fb103f8f Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 18:26:15 -0400 Subject: [PATCH 10/18] Revert "fix" This reverts commit c9ba90571ea096840aa8d11435fb756385adcb8a. --- .jenkins/build.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 5b55ffd980b..3a82a663e31 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -27,8 +27,7 @@ pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1 # Install torchvision from conda pip uninstall -y torchvision || true pip uninstall -y torchvision || true -# GPU, requires CUDA version 9.0 -conda install https://anaconda.org/pytorch/torchvision/0.3.0/download/linux-64/torchvision-0.3.0-py36_cu9.0.176_1.tar.bz2 --no-deps +conda install -c pytorch torchvision --no-deps # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet @@ -45,6 +44,7 @@ if [[ $(pip show torch) ]]; then 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 From 8b1213077b23f40a161cad2725179af53ccd9f05 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 18:26:22 -0400 Subject: [PATCH 11/18] Revert "try to fix" This reverts commit df9a4e4b77d092e0a189b85d315e26edc5ac025f. --- .circleci/config.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index a84b1eef2e8..7125547ca60 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -113,7 +113,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults command: | set -e - export pyTorchDockerImageTag=291 + export pyTorchDockerImageTag=300 echo "PyTorchDockerImageTag: "${pyTorchDockerImageTag} cat >/home/circleci/project/ci_build_script.sh < Date: Thu, 23 May 2019 18:26:29 -0400 Subject: [PATCH 12/18] Revert "try to fix" This reverts commit 49c65a7d0b5ba957cac546236c2000a459917d01. --- .jenkins/build.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 3a82a663e31..0323ad20f99 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -27,7 +27,7 @@ pip install sphinx-gallery tqdm matplotlib ipython pillow==4.1.1 # Install torchvision from conda pip uninstall -y torchvision || true pip uninstall -y torchvision || true -conda install -c pytorch torchvision --no-deps +conda install -c pytorch torchvision # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet From 93b95284ffb7e56b02588c2b65dfdb35e94ea752 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 18:26:35 -0400 Subject: [PATCH 13/18] Revert "update" This reverts commit ac0cc5b81aadae6eaefb3cdf8d7a16fca29b7ecf. --- .circleci/config.yml | 2 +- .jenkins/build.sh | 7 +++---- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 7125547ca60..a84b1eef2e8 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -113,7 +113,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults command: | set -e - export pyTorchDockerImageTag=300 + export pyTorchDockerImageTag=291 echo "PyTorchDockerImageTag: "${pyTorchDockerImageTag} cat >/home/circleci/project/ci_build_script.sh < Date: Thu, 23 May 2019 18:26:47 -0400 Subject: [PATCH 14/18] Revert "Fix torchvision install" This reverts commit 2eca83c623a1441cb3578ecc44adc929edbff800. --- .jenkins/build.sh | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 73217bdf07c..dc8431c9dda 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -25,8 +25,11 @@ 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 pip -pip install torchvision --no-deps +# Install torchvision from source +git clone https://github.com/pytorch/vision --quiet +pushd vision +pip install . --no-deps # We don't want it to install the stock PyTorch version from pip +popd # Install torchaudio from source git clone https://github.com/pytorch/audio --quiet From 31ac199440b27174968915060941aa92a755ebf1 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 18:27:26 -0400 Subject: [PATCH 15/18] try another way --- .jenkins/build.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index dc8431c9dda..115473c92b1 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -28,6 +28,7 @@ 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 From 34405aea49d42f475b140e06cefb71922d79df88 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 22:10:38 -0400 Subject: [PATCH 16/18] try to fix --- .circleci/config.yml | 6 +++--- .jenkins/build.sh | 24 +++++++----------------- 2 files changed, 10 insertions(+), 20 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index a84b1eef2e8..cb0e717a445 100644 --- a/.circleci/config.yml +++ b/.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 diff --git a/.jenkins/build.sh b/.jenkins/build.sh index 115473c92b1..ce8e4925c98 100755 --- a/.jenkins/build.sh +++ b/.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 From def5653d03926f3be2831cbfabc8c4abaf7c3347 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Thu, 23 May 2019 23:48:54 -0400 Subject: [PATCH 17/18] try to fix --- .jenkins/build.sh | 9 --------- requirements.txt | 1 + 2 files changed, 1 insertion(+), 9 deletions(-) diff --git a/.jenkins/build.sh b/.jenkins/build.sh index ce8e4925c98..eaed1d00057 100755 --- a/.jenkins/build.sh +++ b/.jenkins/build.sh @@ -14,7 +14,6 @@ sudo apt-get install -y --no-install-recommends unzip p7zip-full sox libsox-dev export PATH=/opt/conda/bin:$PATH rm -rf src pip install -r $DIR/../requirements.txt -pip uninstall -y torchvision || true export PATH=/opt/conda/bin:$PATH conda install -y sphinx==1.8.2 pandas @@ -25,14 +24,6 @@ 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 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 pushd audio diff --git a/requirements.txt b/requirements.txt index ea0c6fe184e..7d74d4d845e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,6 +5,7 @@ sphinx-gallery tqdm numpy matplotlib +torch torchvision PyHamcrest bs4 From fbaa01d4ab8a02742b3365e006181ab4a40c7e39 Mon Sep 17 00:00:00 2001 From: Will Feng Date: Fri, 24 May 2019 10:32:08 -0400 Subject: [PATCH 18/18] try something different --- .circleci/config.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index cb0e717a445..a84b1eef2e8 100644 --- a/.circleci/config.yml +++ b/.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-cuda10-cudnn7-py3-gcc7" - CUDA_VERSION: "10" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda9-cudnn7-py3" + CUDA_VERSION: "9" 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-cuda10-cudnn7-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda9-cudnn7-py3" resource_class: medium <<: *pytorch_tutorial_build_defaults