pytorch 1.7 support

Summary: CircleCI build configuration to support pytorch1.7, including binaries with cuda 11.0. Note that the default torch on pip is still on cuda 10.2, so I have left the `main` (non conda build) on cuda 10.2 with the existing driver.

Reviewed By: gkioxari

Differential Revision: D24623523

fbshipit-source-id: 59cfa1be06c16225f0f12ed336c07220e8a9a511

.circleci/config.yml

@@ -18,9 +18,7 @@ setupcuda: &setupcuda
     working_directory: ~/
     command: |
       # download and install nvidia drivers, cuda, etc
-      wget --no-verbose --no-clobber -P ~/nvidia-downloads 'https://s3.amazonaws.com/ossci-linux/nvidia_driver/NVIDIA-Linux-x86_64-430.40.run'
       wget --no-verbose --no-clobber -P ~/nvidia-downloads http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
-      sudo /bin/bash ~/nvidia-downloads/NVIDIA-Linux-x86_64-430.40.run --no-drm -q --ui=none
       sudo sh ~/nvidia-downloads/cuda_10.2.89_440.33.01_linux.run --silent
       echo "Done installing CUDA."
       pyenv versions
@@ -157,8 +155,8 @@ jobs:
           sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit=${NVIDIA_CONTAINER_VERSION}
           sudo systemctl restart docker
 
-          DRIVER_FN="NVIDIA-Linux-x86_64-440.59.run"
-          wget "https://s3.amazonaws.com/ossci-linux/nvidia_driver/$DRIVER_FN"
+          DRIVER_FN="NVIDIA-Linux-x86_64-450.80.02.run"
+          wget "https://us.download.nvidia.com/XFree86/Linux-x86_64/450.80.02/$DRIVER_FN"
           sudo /bin/bash "$DRIVER_FN" -s --no-drm || (sudo cat /var/log/nvidia-installer.log && false)
           nvidia-smi
 
@@ -178,7 +176,7 @@ jobs:
           cd ${HOME}/project/
 
           export DOCKER_IMAGE=pytorch/conda-cuda
-          export VARS_TO_PASS="-e PYTHON_VERSION -e BUILD_VERSION -e PYTORCH_VERSION -e UNICODE_ABI -e CU_VERSION"
+          export VARS_TO_PASS="-e PYTHON_VERSION -e BUILD_VERSION -e PYTORCH_VERSION -e CU_VERSION"
 
           docker run --gpus all  --ipc=host -v $(pwd):/remote -w /remote ${VARS_TO_PASS} ${DOCKER_IMAGE} ./packaging/build_conda.sh
 
@@ -260,6 +258,21 @@ workflows:
           name: linux_conda_py36_cu102_pyt160
           python_version: '3.6'
           pytorch_version: 1.6.0
+      - binary_linux_conda:
+          cu_version: cu101
+          name: linux_conda_py36_cu101_pyt170
+          python_version: '3.6'
+          pytorch_version: 1.7.0
+      - binary_linux_conda:
+          cu_version: cu102
+          name: linux_conda_py36_cu102_pyt170
+          python_version: '3.6'
+          pytorch_version: 1.7.0
+      - binary_linux_conda:
+          cu_version: cu110
+          name: linux_conda_py36_cu110_pyt170
+          python_version: '3.6'
+          pytorch_version: 1.7.0
       - binary_linux_conda:
           cu_version: cu92
           name: linux_conda_py37_cu92_pyt14
@@ -315,6 +328,21 @@ workflows:
           name: linux_conda_py37_cu102_pyt160
           python_version: '3.7'
           pytorch_version: 1.6.0
+      - binary_linux_conda:
+          cu_version: cu101
+          name: linux_conda_py37_cu101_pyt170
+          python_version: '3.7'
+          pytorch_version: 1.7.0
+      - binary_linux_conda:
+          cu_version: cu102
+          name: linux_conda_py37_cu102_pyt170
+          python_version: '3.7'
+          pytorch_version: 1.7.0
+      - binary_linux_conda:
+          cu_version: cu110
+          name: linux_conda_py37_cu110_pyt170
+          python_version: '3.7'
+          pytorch_version: 1.7.0
       - binary_linux_conda:
           cu_version: cu92
           name: linux_conda_py38_cu92_pyt14
@@ -370,6 +398,21 @@ workflows:
           name: linux_conda_py38_cu102_pyt160
           python_version: '3.8'
           pytorch_version: 1.6.0
+      - binary_linux_conda:
+          cu_version: cu101
+          name: linux_conda_py38_cu101_pyt170
+          python_version: '3.8'
+          pytorch_version: 1.7.0
+      - binary_linux_conda:
+          cu_version: cu102
+          name: linux_conda_py38_cu102_pyt170
+          python_version: '3.8'
+          pytorch_version: 1.7.0
+      - binary_linux_conda:
+          cu_version: cu110
+          name: linux_conda_py38_cu110_pyt170
+          python_version: '3.8'
+          pytorch_version: 1.7.0
       - binary_linux_wheel:
           cu_version: cu101
           name: linux_wheel_py36_cu101_pyt160
@@ -395,6 +438,11 @@ workflows:
           python_version: "3.7"
           pytorch_version: '1.6.0'
           cu_version: "cu102"
+      - binary_linux_conda_cuda:
+          name: testrun_conda_cuda_py37_cu110_pyt170
+          python_version: "3.7"
+          pytorch_version: '1.7.0'
+          cu_version: "cu110"
       - binary_macos_wheel:
           cu_version: cpu
           name: macos_wheel_py36_cpu