From d049cd2e0102a6b2e08bf2b595131d5177638081 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Mon, 13 Dec 2021 10:10:02 -0800 Subject: [PATCH] PyTorch 1.10 + CUDA 11.1 builds Summary: Although the PyTorch website, which describes the current version 1.10, suggests CUDA 10.2 and 11.3 are supported, it would appear that we need to include builds for CUDA 11.1 to avoid surprises. This is because these builds are on anaconda, and this combination is used on Google Colab. Reviewed By: nikhilaravi Differential Revision: D33063932 fbshipit-source-id: 1b22d1f06e22bd18fb53ceecb58e78ac6a5d1693 --- .circleci/config.yml | 24 ++++++++++++++++++++++++ .circleci/regenerate.py | 2 +- 2 files changed, 25 insertions(+), 1 deletion(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 268a2961..5f62e510 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -340,6 +340,12 @@ workflows: name: linux_conda_py36_cu102_pyt1100 python_version: '3.6' pytorch_version: 1.10.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu111 + name: linux_conda_py36_cu111_pyt1100 + python_version: '3.6' + pytorch_version: 1.10.0 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda113 context: DOCKERHUB_TOKEN @@ -467,6 +473,12 @@ workflows: name: linux_conda_py37_cu102_pyt1100 python_version: '3.7' pytorch_version: 1.10.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu111 + name: linux_conda_py37_cu111_pyt1100 + python_version: '3.7' + pytorch_version: 1.10.0 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda113 context: DOCKERHUB_TOKEN @@ -594,6 +606,12 @@ workflows: name: linux_conda_py38_cu102_pyt1100 python_version: '3.8' pytorch_version: 1.10.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu111 + name: linux_conda_py38_cu111_pyt1100 + python_version: '3.8' + pytorch_version: 1.10.0 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda113 context: DOCKERHUB_TOKEN @@ -685,6 +703,12 @@ workflows: name: linux_conda_py39_cu102_pyt1100 python_version: '3.9' pytorch_version: 1.10.0 + - binary_linux_conda: + context: DOCKERHUB_TOKEN + cu_version: cu111 + name: linux_conda_py39_cu111_pyt1100 + python_version: '3.9' + pytorch_version: 1.10.0 - binary_linux_conda: conda_docker_image: pytorch/conda-builder:cuda113 context: DOCKERHUB_TOKEN diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index d727c882..f93def65 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -26,7 +26,7 @@ CONDA_CUDA_VERSIONS = { "1.8.1": ["cu101", "cu102", "cu111"], "1.9.0": ["cu102", "cu111"], "1.9.1": ["cu102", "cu111"], - "1.10.0": ["cu102", "cu113"], + "1.10.0": ["cu102", "cu111", "cu113"], }