Skip to content

Commit d049cd2

Browse files
bottlerfacebook-github-bot
authored andcommitted
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
1 parent 1edc624 commit d049cd2

File tree

2 files changed

+25
-1
lines changed

2 files changed

+25
-1
lines changed

.circleci/config.yml

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,12 @@ workflows:
340340
name: linux_conda_py36_cu102_pyt1100
341341
python_version: '3.6'
342342
pytorch_version: 1.10.0
343+
- binary_linux_conda:
344+
context: DOCKERHUB_TOKEN
345+
cu_version: cu111
346+
name: linux_conda_py36_cu111_pyt1100
347+
python_version: '3.6'
348+
pytorch_version: 1.10.0
343349
- binary_linux_conda:
344350
conda_docker_image: pytorch/conda-builder:cuda113
345351
context: DOCKERHUB_TOKEN
@@ -467,6 +473,12 @@ workflows:
467473
name: linux_conda_py37_cu102_pyt1100
468474
python_version: '3.7'
469475
pytorch_version: 1.10.0
476+
- binary_linux_conda:
477+
context: DOCKERHUB_TOKEN
478+
cu_version: cu111
479+
name: linux_conda_py37_cu111_pyt1100
480+
python_version: '3.7'
481+
pytorch_version: 1.10.0
470482
- binary_linux_conda:
471483
conda_docker_image: pytorch/conda-builder:cuda113
472484
context: DOCKERHUB_TOKEN
@@ -594,6 +606,12 @@ workflows:
594606
name: linux_conda_py38_cu102_pyt1100
595607
python_version: '3.8'
596608
pytorch_version: 1.10.0
609+
- binary_linux_conda:
610+
context: DOCKERHUB_TOKEN
611+
cu_version: cu111
612+
name: linux_conda_py38_cu111_pyt1100
613+
python_version: '3.8'
614+
pytorch_version: 1.10.0
597615
- binary_linux_conda:
598616
conda_docker_image: pytorch/conda-builder:cuda113
599617
context: DOCKERHUB_TOKEN
@@ -685,6 +703,12 @@ workflows:
685703
name: linux_conda_py39_cu102_pyt1100
686704
python_version: '3.9'
687705
pytorch_version: 1.10.0
706+
- binary_linux_conda:
707+
context: DOCKERHUB_TOKEN
708+
cu_version: cu111
709+
name: linux_conda_py39_cu111_pyt1100
710+
python_version: '3.9'
711+
pytorch_version: 1.10.0
688712
- binary_linux_conda:
689713
conda_docker_image: pytorch/conda-builder:cuda113
690714
context: DOCKERHUB_TOKEN

.circleci/regenerate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
"1.8.1": ["cu101", "cu102", "cu111"],
2727
"1.9.0": ["cu102", "cu111"],
2828
"1.9.1": ["cu102", "cu111"],
29-
"1.10.0": ["cu102", "cu113"],
29+
"1.10.0": ["cu102", "cu111", "cu113"],
3030
}
3131

3232

0 commit comments

Comments
 (0)