diff --git a/.circleci/config.yml b/.circleci/config.yml index a5be618c982..b1edadbd6e8 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -81,6 +81,93 @@ jobs: paths: - "*" + binary_linux_conda_cuda: + <<: *binary_common + machine: + image: ubuntu-1604:201903-01 + resource_class: gpu.medium + steps: + - checkout + - run: + name: Build + command: | + set -e + + curl -L https://packagecloud.io/circleci/trusty/gpgkey | sudo apt-key add - + curl -L https://dl.google.com/linux/linux_signing_key.pub | sudo apt-key add - + + sudo apt-get update + + sudo apt-get install \ + apt-transport-https \ + ca-certificates \ + curl \ + gnupg-agent \ + software-properties-common + + curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add - + + sudo add-apt-repository \ + "deb [arch=amd64] https://download.docker.com/linux/ubuntu \ + $(lsb_release -cs) \ + stable" + + sudo apt-get update + sudo apt-get install docker-ce docker-ce-cli containerd.io + + # Add the package repositories + distribution=$(. /etc/os-release;echo $ID$VERSION_ID) + curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - + curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list + + sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit + sudo systemctl restart docker + + + DRIVER_FN="NVIDIA-Linux-x86_64-410.104.run" + wget "https://s3.amazonaws.com/ossci-linux/nvidia_driver/$DRIVER_FN" + sudo /bin/bash "$DRIVER_FN" -s --no-drm || (sudo cat /var/log/nvidia-installer.log && false) + nvidia-smi + + # export DOCKER_IMAGE=nvidia/cuda:10.0-base + export DOCKER_IMAGE=soumith/conda-cuda + docker pull $DOCKER_IMAGE >/dev/null + + export id=$(docker run --gpus all -t -d ${DOCKER_IMAGE}) + + docker cp /home/circleci/project/. $id:/workspace + + cat >/home/circleci/install_and_test.sh \</dev/null + + export id=$(docker run --gpus all -t -d ${DOCKER_IMAGE}) + + docker cp /home/circleci/project/. $id:/workspace + + cat >/home/circleci/install_and_test.sh \<