diff --git a/.circleci/config.yml b/.circleci/config.yml index 3564824a7dc..176fe52f5b9 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -134,6 +134,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults fi set -x + echo 'rm /opt/cache/bin/*' | docker exec -u root -i "$id" bash docker cp /home/circleci/project/. "$id:/var/lib/jenkins/workspace" export COMMAND='((echo "source ./workspace/env" && echo "sudo chown -R jenkins workspace && cd workspace && ./ci_build_script.sh") | docker exec -u jenkins -i "$id" bash) 2>&1' @@ -141,14 +142,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.2-cudnn7-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7" CUDA_VERSION: "9" resource_class: gpu.nvidia.small <<: *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.2-cudnn7-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7" resource_class: medium <<: *pytorch_tutorial_build_defaults diff --git a/.circleci/config.yml.in b/.circleci/config.yml.in index 215571a82f1..c5a4e1266f2 100644 --- a/.circleci/config.yml.in +++ b/.circleci/config.yml.in @@ -134,6 +134,7 @@ pytorch_tutorial_build_defaults: &pytorch_tutorial_build_defaults fi set -x + echo 'rm /opt/cache/bin/*' | docker exec -u root -i "$id" bash docker cp /home/circleci/project/. "$id:/var/lib/jenkins/workspace" export COMMAND='((echo "source ./workspace/env" && echo "sudo chown -R jenkins workspace && cd workspace && ./ci_build_script.sh") | docker exec -u jenkins -i "$id" bash) 2>&1' @@ -141,14 +142,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.2-cudnn7-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7" CUDA_VERSION: "9" resource_class: gpu.nvidia.small <<: *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.2-cudnn7-py3-gcc7" + DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-cuda11.6-cudnn8-py3-gcc7" resource_class: medium <<: *pytorch_tutorial_build_defaults {% raw %}