Do not install nvidia docker for non-NVIDIA configs (#41144)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/41144
Differential Revision: D22457124
Pulled By: malfet
fbshipit-source-id: e615199cb78b315aa700efcc7332ebf4299212bf
diff --git a/.circleci/scripts/setup_ci_environment.sh b/.circleci/scripts/setup_ci_environment.sh
index 189ad64..b0c207e 100755
--- a/.circleci/scripts/setup_ci_environment.sh
+++ b/.circleci/scripts/setup_ci_environment.sh
@@ -2,10 +2,12 @@
set -ex -o pipefail
# Set up NVIDIA docker repo
-curl -s -L --retry 3 https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
-echo "deb https://nvidia.github.io/libnvidia-container/ubuntu16.04/amd64 /" | sudo tee -a /etc/apt/sources.list.d/nvidia-docker.list
-echo "deb https://nvidia.github.io/nvidia-container-runtime/ubuntu16.04/amd64 /" | sudo tee -a /etc/apt/sources.list.d/nvidia-docker.list
-echo "deb https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64 /" | sudo tee -a /etc/apt/sources.list.d/nvidia-docker.list
+if [[ "${BUILD_ENVIRONMENT}" == *cuda* ]]; then
+ curl -s -L --retry 3 https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
+ echo "deb https://nvidia.github.io/libnvidia-container/ubuntu16.04/amd64 /" | sudo tee -a /etc/apt/sources.list.d/nvidia-docker.list
+ echo "deb https://nvidia.github.io/nvidia-container-runtime/ubuntu16.04/amd64 /" | sudo tee -a /etc/apt/sources.list.d/nvidia-docker.list
+ echo "deb https://nvidia.github.io/nvidia-docker/ubuntu16.04/amd64 /" | sudo tee -a /etc/apt/sources.list.d/nvidia-docker.list
+fi
# Remove unnecessary sources
sudo rm -f /etc/apt/sources.list.d/google-chrome.list
@@ -40,14 +42,21 @@
# Using 'retry' here as belt-and-suspenders even though we are
# presumably retrying at the single-package level via the
# apt.conf.d/80-retries technique.
-retry sudo apt-get -y install \
- linux-headers-$(uname -r) \
- linux-image-generic \
- moreutils \
- docker-ce=5:18.09.4~3-0~ubuntu-xenial \
- nvidia-container-runtime=2.0.0+docker18.09.4-1 \
- nvidia-docker2=2.0.3+docker18.09.4-1 \
- expect-dev
+if [[ "${BUILD_ENVIRONMENT}" == *cuda* ]]; then
+ retry sudo apt-get -y install \
+ linux-headers-$(uname -r) \
+ linux-image-generic \
+ moreutils \
+ docker-ce=5:18.09.4~3-0~ubuntu-xenial \
+ nvidia-container-runtime=2.0.0+docker18.09.4-1 \
+ nvidia-docker2=2.0.3+docker18.09.4-1 \
+ expect-dev
+else
+ retry sudo apt-get -y install \
+ moreutils \
+ docker-ce=5:18.09.4~3-0~ubuntu-xenial \
+ expect-dev
+fi
sudo pkill -SIGHUP dockerd
