builds for PyTorch 1.9.1

Summary: Add conda builds for the newly released PyTorch version 1.9.1.

Reviewed By: patricklabatut

Differential Revision: D31140206

fbshipit-source-id: 697549a3ef0db8248f4f9b5c00cf1407296b5022
This commit is contained in:
Jeremy Reizenstein 2021-09-27 04:15:53 -07:00 committed by Facebook GitHub Bot
parent 9a737da83c
commit d0ca3b9e0c
3 changed files with 50 additions and 0 deletions

View File

@ -368,6 +368,18 @@ workflows:
name: linux_conda_py36_cu111_pyt190 name: linux_conda_py36_cu111_pyt190
python_version: '3.6' python_version: '3.6'
pytorch_version: 1.9.0 pytorch_version: 1.9.0
- binary_linux_conda:
context: DOCKERHUB_TOKEN
cu_version: cu102
name: linux_conda_py36_cu102_pyt191
python_version: '3.6'
pytorch_version: 1.9.1
- binary_linux_conda:
context: DOCKERHUB_TOKEN
cu_version: cu111
name: linux_conda_py36_cu111_pyt191
python_version: '3.6'
pytorch_version: 1.9.1
- binary_linux_conda: - binary_linux_conda:
context: DOCKERHUB_TOKEN context: DOCKERHUB_TOKEN
cu_version: cu92 cu_version: cu92
@ -506,6 +518,18 @@ workflows:
name: linux_conda_py37_cu111_pyt190 name: linux_conda_py37_cu111_pyt190
python_version: '3.7' python_version: '3.7'
pytorch_version: 1.9.0 pytorch_version: 1.9.0
- binary_linux_conda:
context: DOCKERHUB_TOKEN
cu_version: cu102
name: linux_conda_py37_cu102_pyt191
python_version: '3.7'
pytorch_version: 1.9.1
- binary_linux_conda:
context: DOCKERHUB_TOKEN
cu_version: cu111
name: linux_conda_py37_cu111_pyt191
python_version: '3.7'
pytorch_version: 1.9.1
- binary_linux_conda: - binary_linux_conda:
context: DOCKERHUB_TOKEN context: DOCKERHUB_TOKEN
cu_version: cu92 cu_version: cu92
@ -644,6 +668,18 @@ workflows:
name: linux_conda_py38_cu111_pyt190 name: linux_conda_py38_cu111_pyt190
python_version: '3.8' python_version: '3.8'
pytorch_version: 1.9.0 pytorch_version: 1.9.0
- binary_linux_conda:
context: DOCKERHUB_TOKEN
cu_version: cu102
name: linux_conda_py38_cu102_pyt191
python_version: '3.8'
pytorch_version: 1.9.1
- binary_linux_conda:
context: DOCKERHUB_TOKEN
cu_version: cu111
name: linux_conda_py38_cu111_pyt191
python_version: '3.8'
pytorch_version: 1.9.1
- binary_linux_conda: - binary_linux_conda:
context: DOCKERHUB_TOKEN context: DOCKERHUB_TOKEN
cu_version: cu101 cu_version: cu101
@ -710,6 +746,18 @@ workflows:
name: linux_conda_py39_cu111_pyt190 name: linux_conda_py39_cu111_pyt190
python_version: '3.9' python_version: '3.9'
pytorch_version: 1.9.0 pytorch_version: 1.9.0
- binary_linux_conda:
context: DOCKERHUB_TOKEN
cu_version: cu102
name: linux_conda_py39_cu102_pyt191
python_version: '3.9'
pytorch_version: 1.9.1
- binary_linux_conda:
context: DOCKERHUB_TOKEN
cu_version: cu111
name: linux_conda_py39_cu111_pyt191
python_version: '3.9'
pytorch_version: 1.9.1
- binary_linux_conda_cuda: - binary_linux_conda_cuda:
name: testrun_conda_cuda_py36_cu101_pyt151 name: testrun_conda_cuda_py36_cu101_pyt151
context: DOCKERHUB_TOKEN context: DOCKERHUB_TOKEN

View File

@ -27,6 +27,7 @@ CONDA_CUDA_VERSIONS = {
"1.8.0": ["cu101", "cu102", "cu111"], "1.8.0": ["cu101", "cu102", "cu111"],
"1.8.1": ["cu101", "cu102", "cu111"], "1.8.1": ["cu101", "cu102", "cu111"],
"1.9.0": ["cu102", "cu111"], "1.9.0": ["cu102", "cu111"],
"1.9.1": ["cu102", "cu111"],
} }

View File

@ -38,6 +38,7 @@ declare -A CONDA_CUDA_VERSIONS=(
# ["1.8.0"]="cu101 cu102 cu111" # ["1.8.0"]="cu101 cu102 cu111"
# ["1.8.1"]="cu101 cu102 cu111" # ["1.8.1"]="cu101 cu102 cu111"
["1.9.0"]="cu102 cu111" ["1.9.0"]="cu102 cu111"
["1.9.1"]="cu102 cu111"
) )