Skip to content

GitLab

Unverified Commit 0e32a12c authored by Eli Uriegas's avatar Eli Uriegas Committed by GitHub
Browse files

.circleci: Downgrade CUDA 11.2 -> 11.1 (#3418)

parent 433283eb
Show whitespace changes
Inline Side-by-side
Showing with 214 additions and 175 deletions
.circleci/config.yml
View file @ 0e32a12c
......@@ -216,6 +216,7 @@ jobs:
- designate_upload_channel
- run:
name: Build conda packages
no_output_timeout: 20m
command: |
set -ex
source packaging/windows/internal/vc_install_helper.sh
......@@ -825,11 +826,11 @@ workflows:
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
name: binary_linux_wheel_py3.6_cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
name: binary_linux_wheel_py3.6_cu111
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -849,11 +850,11 @@ workflows:
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
name: binary_linux_wheel_py3.7_cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
name: binary_linux_wheel_py3.7_cu111
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -873,11 +874,11 @@ workflows:
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
name: binary_linux_wheel_py3.8_cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
name: binary_linux_wheel_py3.8_cu111
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -897,11 +898,11 @@ workflows:
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
name: binary_linux_wheel_py3.9_cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
name: binary_linux_wheel_py3.9_cu111
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -954,13 +955,13 @@ workflows:
name: binary_win_wheel_py3.6_cu102
python_version: '3.6'
- binary_win_wheel:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.6_cu112
name: binary_win_wheel_py3.6_cu111
python_version: '3.6'
- binary_win_wheel:
cu_version: cpu
......@@ -990,13 +991,13 @@ workflows:
name: binary_win_wheel_py3.7_cu102
python_version: '3.7'
- binary_win_wheel:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.7_cu112
name: binary_win_wheel_py3.7_cu111
python_version: '3.7'
- binary_win_wheel:
cu_version: cpu
......@@ -1026,13 +1027,13 @@ workflows:
name: binary_win_wheel_py3.8_cu102
python_version: '3.8'
- binary_win_wheel:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.8_cu112
name: binary_win_wheel_py3.8_cu111
python_version: '3.8'
- binary_win_wheel:
cu_version: cpu
......@@ -1057,8 +1058,8 @@ workflows:
name: binary_win_wheel_py3.9_cu102
python_version: '3.9'
- binary_win_wheel:
cu_version: cu112
name: binary_win_wheel_py3.9_cu112
cu_version: cu111
name: binary_win_wheel_py3.9_cu111
python_version: '3.9'
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
......@@ -1079,11 +1080,11 @@ workflows:
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
name: binary_linux_conda_py3.6_cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
name: binary_linux_conda_py3.6_cu111
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1103,11 +1104,11 @@ workflows:
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
name: binary_linux_conda_py3.7_cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
name: binary_linux_conda_py3.7_cu111
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1127,11 +1128,11 @@ workflows:
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
name: binary_linux_conda_py3.8_cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
name: binary_linux_conda_py3.8_cu111
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1151,11 +1152,11 @@ workflows:
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
name: binary_linux_conda_py3.9_cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
name: binary_linux_conda_py3.9_cu111
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1208,13 +1209,13 @@ workflows:
name: binary_win_conda_py3.6_cu102
python_version: '3.6'
- binary_win_conda:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.6_cu112
name: binary_win_conda_py3.6_cu111
python_version: '3.6'
- binary_win_conda:
cu_version: cpu
......@@ -1244,13 +1245,13 @@ workflows:
name: binary_win_conda_py3.7_cu102
python_version: '3.7'
- binary_win_conda:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.7_cu112
name: binary_win_conda_py3.7_cu111
python_version: '3.7'
- binary_win_conda:
cu_version: cpu
......@@ -1280,13 +1281,13 @@ workflows:
name: binary_win_conda_py3.8_cu102
python_version: '3.8'
- binary_win_conda:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.8_cu112
name: binary_win_conda_py3.8_cu111
python_version: '3.8'
- binary_win_conda:
cu_version: cpu
......@@ -1311,8 +1312,8 @@ workflows:
name: binary_win_conda_py3.9_cu102
python_version: '3.9'
- binary_win_conda:
cu_version: cu112
name: binary_win_conda_py3.9_cu112
cu_version: cu111
name: binary_win_conda_py3.9_cu111
python_version: '3.9'
- build_docs:
name: build_docs
......@@ -1578,16 +1579,16 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.6_cu102_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_cu112
name: nightly_binary_linux_wheel_py3.6_cu111
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload:
context: org-member
filters:
......@@ -1595,19 +1596,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_cu112_upload
name: nightly_binary_linux_wheel_py3.6_cu111_upload
requires:
- nightly_binary_linux_wheel_py3.6_cu112
subfolder: cu112/
- nightly_binary_linux_wheel_py3.6_cu111
subfolder: cu111/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.6_cu112_smoke_test_pip
name: nightly_binary_linux_wheel_py3.6_cu111_smoke_test_pip
python_version: '3.6'
requires:
- nightly_binary_linux_wheel_py3.6_cu112_upload
- nightly_binary_linux_wheel_py3.6_cu111_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1702,16 +1703,16 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.7_cu102_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu112
name: nightly_binary_linux_wheel_py3.7_cu111
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload:
context: org-member
filters:
......@@ -1719,19 +1720,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu112_upload
name: nightly_binary_linux_wheel_py3.7_cu111_upload
requires:
- nightly_binary_linux_wheel_py3.7_cu112
subfolder: cu112/
- nightly_binary_linux_wheel_py3.7_cu111
subfolder: cu111/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.7_cu112_smoke_test_pip
name: nightly_binary_linux_wheel_py3.7_cu111_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_cu112_upload
- nightly_binary_linux_wheel_py3.7_cu111_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1826,16 +1827,16 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.8_cu102_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu112
name: nightly_binary_linux_wheel_py3.8_cu111
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload:
context: org-member
filters:
......@@ -1843,19 +1844,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu112_upload
name: nightly_binary_linux_wheel_py3.8_cu111_upload
requires:
- nightly_binary_linux_wheel_py3.8_cu112
subfolder: cu112/
- nightly_binary_linux_wheel_py3.8_cu111
subfolder: cu111/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.8_cu112_smoke_test_pip
name: nightly_binary_linux_wheel_py3.8_cu111_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_cu112_upload
- nightly_binary_linux_wheel_py3.8_cu111_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -1950,16 +1951,16 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.9_cu102_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu112
name: nightly_binary_linux_wheel_py3.9_cu111
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload:
context: org-member
filters:
......@@ -1967,19 +1968,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu112_upload
name: nightly_binary_linux_wheel_py3.9_cu111_upload
requires:
- nightly_binary_linux_wheel_py3.9_cu112
subfolder: cu112/
- nightly_binary_linux_wheel_py3.9_cu111
subfolder: cu111/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.9_cu112_smoke_test_pip
name: nightly_binary_linux_wheel_py3.9_cu111_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_cu112_upload
- nightly_binary_linux_wheel_py3.9_cu111_upload
- binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2156,13 +2157,13 @@ workflows:
requires:
- nightly_binary_win_wheel_py3.6_cu102_upload
- binary_win_wheel:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.6_cu112
name: nightly_binary_win_wheel_py3.6_cu111
python_version: '3.6'
- binary_wheel_upload:
context: org-member
......@@ -2171,19 +2172,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.6_cu112_upload
name: nightly_binary_win_wheel_py3.6_cu111_upload
requires:
- nightly_binary_win_wheel_py3.6_cu112
subfolder: cu112/
- nightly_binary_win_wheel_py3.6_cu111
subfolder: cu111/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.6_cu112_smoke_test_pip
name: nightly_binary_win_wheel_py3.6_cu111_smoke_test_pip
python_version: '3.6'
requires:
- nightly_binary_win_wheel_py3.6_cu112_upload
- nightly_binary_win_wheel_py3.6_cu111_upload
- binary_win_wheel:
cu_version: cpu
filters:
......@@ -2272,13 +2273,13 @@ workflows:
requires:
- nightly_binary_win_wheel_py3.7_cu102_upload
- binary_win_wheel:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.7_cu112
name: nightly_binary_win_wheel_py3.7_cu111
python_version: '3.7'
- binary_wheel_upload:
context: org-member
......@@ -2287,19 +2288,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.7_cu112_upload
name: nightly_binary_win_wheel_py3.7_cu111_upload
requires:
- nightly_binary_win_wheel_py3.7_cu112
subfolder: cu112/
- nightly_binary_win_wheel_py3.7_cu111
subfolder: cu111/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.7_cu112_smoke_test_pip
name: nightly_binary_win_wheel_py3.7_cu111_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_win_wheel_py3.7_cu112_upload
- nightly_binary_win_wheel_py3.7_cu111_upload
- binary_win_wheel:
cu_version: cpu
filters:
......@@ -2388,13 +2389,13 @@ workflows:
requires:
- nightly_binary_win_wheel_py3.8_cu102_upload
- binary_win_wheel:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu112
name: nightly_binary_win_wheel_py3.8_cu111
python_version: '3.8'
- binary_wheel_upload:
context: org-member
......@@ -2403,19 +2404,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu112_upload
name: nightly_binary_win_wheel_py3.8_cu111_upload
requires:
- nightly_binary_win_wheel_py3.8_cu112
subfolder: cu112/
- nightly_binary_win_wheel_py3.8_cu111
subfolder: cu111/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.8_cu112_smoke_test_pip
name: nightly_binary_win_wheel_py3.8_cu111_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_win_wheel_py3.8_cu112_upload
- nightly_binary_win_wheel_py3.8_cu111_upload
- binary_win_wheel:
cu_version: cpu
filters:
......@@ -2504,13 +2505,13 @@ workflows:
requires:
- nightly_binary_win_wheel_py3.9_cu102_upload
- binary_win_wheel:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.9_cu112
name: nightly_binary_win_wheel_py3.9_cu111
python_version: '3.9'
- binary_wheel_upload:
context: org-member
......@@ -2519,19 +2520,19 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.9_cu112_upload
name: nightly_binary_win_wheel_py3.9_cu111_upload
requires:
- nightly_binary_win_wheel_py3.9_cu112
subfolder: cu112/
- nightly_binary_win_wheel_py3.9_cu111
subfolder: cu111/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.9_cu112_smoke_test_pip
name: nightly_binary_win_wheel_py3.9_cu111_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_win_wheel_py3.9_cu112_upload
- nightly_binary_win_wheel_py3.9_cu111_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2623,16 +2624,16 @@ workflows:
requires:
- nightly_binary_linux_conda_py3.6_cu102_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_cu112
name: nightly_binary_linux_conda_py3.6_cu111
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_conda_upload:
context: org-member
filters:
......@@ -2640,18 +2641,18 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_cu112_upload
name: nightly_binary_linux_conda_py3.6_cu111_upload
requires:
- nightly_binary_linux_conda_py3.6_cu112
- nightly_binary_linux_conda_py3.6_cu111
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.6_cu112_smoke_test_conda
name: nightly_binary_linux_conda_py3.6_cu111_smoke_test_conda
python_version: '3.6'
requires:
- nightly_binary_linux_conda_py3.6_cu112_upload
- nightly_binary_linux_conda_py3.6_cu111_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2743,16 +2744,16 @@ workflows:
requires:
- nightly_binary_linux_conda_py3.7_cu102_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu112
name: nightly_binary_linux_conda_py3.7_cu111
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_conda_upload:
context: org-member
filters:
......@@ -2760,18 +2761,18 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu112_upload
name: nightly_binary_linux_conda_py3.7_cu111_upload
requires:
- nightly_binary_linux_conda_py3.7_cu112
- nightly_binary_linux_conda_py3.7_cu111
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.7_cu112_smoke_test_conda
name: nightly_binary_linux_conda_py3.7_cu111_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_linux_conda_py3.7_cu112_upload
- nightly_binary_linux_conda_py3.7_cu111_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2863,16 +2864,16 @@ workflows:
requires:
- nightly_binary_linux_conda_py3.8_cu102_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu112
name: nightly_binary_linux_conda_py3.8_cu111
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_conda_upload:
context: org-member
filters:
......@@ -2880,18 +2881,18 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu112_upload
name: nightly_binary_linux_conda_py3.8_cu111_upload
requires:
- nightly_binary_linux_conda_py3.8_cu112
- nightly_binary_linux_conda_py3.8_cu111
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.8_cu112_smoke_test_conda
name: nightly_binary_linux_conda_py3.8_cu111_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_linux_conda_py3.8_cu112_upload
- nightly_binary_linux_conda_py3.8_cu111_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -2983,16 +2984,16 @@ workflows:
requires:
- nightly_binary_linux_conda_py3.9_cu102_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112
cu_version: cu112
conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu112
name: nightly_binary_linux_conda_py3.9_cu111
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda112
wheel_docker_image: pytorch/manylinux-cuda111
- binary_conda_upload:
context: org-member
filters:
......@@ -3000,18 +3001,18 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu112_upload
name: nightly_binary_linux_conda_py3.9_cu111_upload
requires:
- nightly_binary_linux_conda_py3.9_cu112
- nightly_binary_linux_conda_py3.9_cu111
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.9_cu112_smoke_test_conda
name: nightly_binary_linux_conda_py3.9_cu111_smoke_test_conda
python_version: '3.9'
requires:
- nightly_binary_linux_conda_py3.9_cu112_upload
- nightly_binary_linux_conda_py3.9_cu111_upload
- binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu
......@@ -3181,13 +3182,13 @@ workflows:
requires:
- nightly_binary_win_conda_py3.6_cu102_upload
- binary_win_conda:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.6_cu112
name: nightly_binary_win_conda_py3.6_cu111
python_version: '3.6'
- binary_conda_upload:
context: org-member
......@@ -3196,18 +3197,18 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.6_cu112_upload
name: nightly_binary_win_conda_py3.6_cu111_upload
requires:
- nightly_binary_win_conda_py3.6_cu112
- nightly_binary_win_conda_py3.6_cu111
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.6_cu112_smoke_test_conda
name: nightly_binary_win_conda_py3.6_cu111_smoke_test_conda
python_version: '3.6'
requires:
- nightly_binary_win_conda_py3.6_cu112_upload
- nightly_binary_win_conda_py3.6_cu111_upload
- binary_win_conda:
cu_version: cpu
filters:
......@@ -3293,13 +3294,13 @@ workflows:
requires:
- nightly_binary_win_conda_py3.7_cu102_upload
- binary_win_conda:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.7_cu112
name: nightly_binary_win_conda_py3.7_cu111
python_version: '3.7'
- binary_conda_upload:
context: org-member
......@@ -3308,18 +3309,18 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.7_cu112_upload
name: nightly_binary_win_conda_py3.7_cu111_upload
requires:
- nightly_binary_win_conda_py3.7_cu112
- nightly_binary_win_conda_py3.7_cu111
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.7_cu112_smoke_test_conda
name: nightly_binary_win_conda_py3.7_cu111_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_win_conda_py3.7_cu112_upload
- nightly_binary_win_conda_py3.7_cu111_upload
- binary_win_conda:
cu_version: cpu
filters:
......@@ -3405,13 +3406,13 @@ workflows:
requires:
- nightly_binary_win_conda_py3.8_cu102_upload
- binary_win_conda:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu112
name: nightly_binary_win_conda_py3.8_cu111
python_version: '3.8'
- binary_conda_upload:
context: org-member
......@@ -3420,18 +3421,18 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu112_upload
name: nightly_binary_win_conda_py3.8_cu111_upload
requires:
- nightly_binary_win_conda_py3.8_cu112
- nightly_binary_win_conda_py3.8_cu111
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.8_cu112_smoke_test_conda
name: nightly_binary_win_conda_py3.8_cu111_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_win_conda_py3.8_cu112_upload
- nightly_binary_win_conda_py3.8_cu111_upload
- binary_win_conda:
cu_version: cpu
filters:
......@@ -3517,13 +3518,13 @@ workflows:
requires:
- nightly_binary_win_conda_py3.9_cu102_upload
- binary_win_conda:
cu_version: cu112
cu_version: cu111
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.9_cu112
name: nightly_binary_win_conda_py3.9_cu111
python_version: '3.9'
- binary_conda_upload:
context: org-member
......@@ -3532,18 +3533,18 @@ workflows:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.9_cu112_upload
name: nightly_binary_win_conda_py3.9_cu111_upload
requires:
- nightly_binary_win_conda_py3.9_cu112
- nightly_binary_win_conda_py3.9_cu111
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.9_cu112_smoke_test_conda
name: nightly_binary_win_conda_py3.9_cu111_smoke_test_conda
python_version: '3.9'
requires:
- nightly_binary_win_conda_py3.9_cu112_upload
- nightly_binary_win_conda_py3.9_cu111_upload
docker_build:
triggers:
- schedule:
......
.circleci/config.yml.in
View file @ 0e32a12c
......@@ -216,6 +216,7 @@ jobs:
- designate_upload_channel
- run:
name: Build conda packages
no_output_timeout: 20m
command: |
set -ex
source packaging/windows/internal/vc_install_helper.sh
......
.circleci/regenerate.py
View file @ 0e32a12c
......@@ -20,7 +20,6 @@ import os.path
PYTHON_VERSIONS = ["3.6", "3.7", "3.8", "3.9"]
CUDA_VERSION = ["10.1", "10.2", "11.2"]
RC_PATTERN = r"/v[0-9]+(\.[0-9]+)*-rc[0-9]+/"
......@@ -30,8 +29,8 @@ def build_workflows(prefix='', filter_branch=None, upload=False, indentation=6,
for btype in ["wheel", "conda"]:
for os_type in ["linux", "macos", "win"]:
python_versions = PYTHON_VERSIONS
cu_versions_dict = {"linux": ["cpu", "cu101", "cu102", "cu112"],
"win": ["cpu", "cu101", "cu102", "cu112"],
cu_versions_dict = {"linux": ["cpu", "cu101", "cu102", "cu111"],
"win": ["cpu", "cu101", "cu102", "cu111"],
"macos": ["cpu"]}
cu_versions = cu_versions_dict[os_type]
for python_version in python_versions:
......@@ -103,6 +102,7 @@ manylinux_images = {
"cu101": "pytorch/manylinux-cuda101",
"cu102": "pytorch/manylinux-cuda102",
"cu110": "pytorch/manylinux-cuda110",
"cu111": "pytorch/manylinux-cuda111",
"cu112": "pytorch/manylinux-cuda112",
}
......
packaging/pkg_helpers.bash
View file @ 0e32a12c
......@@ -60,6 +60,17 @@ setup_cuda() {
# https://github.com/pytorch/pytorch/pull/23408 lands
export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_86,code=sm_86 -gencode=arch=compute_50,code=compute_50"
;;
cu111)
if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.1"
else
export CUDA_HOME=/usr/local/cuda-11.1/
fi
export FORCE_CUDA=1
# Hard-coding gencode flags is temporary situation until
# https://github.com/pytorch/pytorch/pull/23408 lands
export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_86,code=sm_86 -gencode=arch=compute_50,code=compute_50"
;;
cu110)
if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0"
......@@ -293,6 +304,9 @@ setup_conda_cudatoolkit_constraint() {
cu112)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.2,<11.3 # [not osx]"
;;
cu111)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.1,<11.2 # [not osx]"
;;
cu110)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.0,<11.1 # [not osx]"
;;
......@@ -331,6 +345,9 @@ setup_conda_cudatoolkit_plain_constraint() {
cu112)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=11.2"
;;
cu111)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=11.1"
;;
cu102)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=10.2"
;;
......
packaging/torchvision/meta.yaml
View file @ 0e32a12c
......@@ -16,8 +16,7 @@ requirements:
host:
- python
- setuptools
- numpy >=1.20 # [py>=39]
- numpy >=1.11 # [py!=39]
- defaults::numpy >=1.11
{{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT') }}
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
{{ environ.get('CONDA_CPUONLY_FEATURE') }}
......@@ -28,8 +27,7 @@ requirements:
- ffmpeg >=4.2 # [not win]
- jpeg <=9b
- pillow >=4.1.1
- numpy >=1.20 # [py>=39]
- numpy >=1.11 # [py!=39]
- defaults::numpy >=1.11
{{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
......
packaging/windows/internal/cuda_install.bat
View file @ 0e32a12c
......@@ -19,6 +19,7 @@ if %CUDA_VER% EQU 100 goto cuda100
if %CUDA_VER% EQU 101 goto cuda101
if %CUDA_VER% EQU 102 goto cuda102
if %CUDA_VER% EQU 110 goto cuda110
if %CUDA_VER% EQU 111 goto cuda111
if %CUDA_VER% EQU 112 goto cuda112
echo CUDA %CUDA_VERSION_STR% is not supported
......@@ -108,6 +109,27 @@ if not exist "%SRC_DIR%\temp_build\cudnn-11.0-windows-x64-v8.0.4.30.zip" (
goto cuda_common
:cuda111
if not exist "%SRC_DIR%\temp_build\cuda_11.1.0_456.43_win10.exe" (
curl -k -L https://ossci-windows.s3.amazonaws.com/cuda_11.1.0_456.43_win10.exe --output "%SRC_DIR%\temp_build\cuda_11.1.0_456.43_win10.exe"
if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\cuda_11.1.0_456.43_win10.exe"
set "ARGS=nvcc_11.1 cuobjdump_11.1 nvprune_11.1 nvprof_11.1 cupti_11.1 cublas_11.1 cublas_dev_11.1 cudart_11.1 cufft_11.1 cufft_dev_11.1 curand_11.1 curand_dev_11.1 cusolver_11.1 cusolver_dev_11.1 cusparse_11.1 cusparse_dev_11.1 npp_11.1 npp_dev_11.1 nvrtc_11.1 nvrtc_dev_11.1 nvml_dev_11.1"
)
@REM There is no downloadable driver for Tesla on CUDA 11.1 yet. We will use
@REM the driver inside CUDA
if "%JOB_EXECUTOR%" == "windows-with-nvidia-gpu" set "ARGS=%ARGS% Display.Driver"
if not exist "%SRC_DIR%\temp_build\cudnn-11.1-windows-x64-v8.0.5.39.zip" (
curl -k -L https://ossci-windows.s3.amazonaws.com/cudnn-11.1-windows-x64-v8.0.5.39.zip --output "%SRC_DIR%\temp_build\cudnn-11.1-windows-x64-v8.0.5.39.zip"
if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\cudnn-11.1-windows-x64-v8.0.5.39.zip"
)
goto cuda_common
:cuda112
if not exist "%SRC_DIR%\temp_build\cuda_11.2.0_460.89_win10.exe" (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment