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
Hide whitespace changes
Inline Side-by-side
Showing with 214 additions and 175 deletions
.circleci/config.yml
View file @ 0e32a12c
...@@ -216,6 +216,7 @@ jobs: ...@@ -216,6 +216,7 @@ jobs:
- designate_upload_channel - designate_upload_channel
- run: - run:
name: Build conda packages name: Build conda packages
no_output_timeout: 20m
command: | command: |
set -ex set -ex
source packaging/windows/internal/vc_install_helper.sh source packaging/windows/internal/vc_install_helper.sh
...@@ -825,11 +826,11 @@ workflows: ...@@ -825,11 +826,11 @@ workflows:
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
name: binary_linux_wheel_py3.6_cu112 name: binary_linux_wheel_py3.6_cu111
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -849,11 +850,11 @@ workflows: ...@@ -849,11 +850,11 @@ workflows:
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
name: binary_linux_wheel_py3.7_cu112 name: binary_linux_wheel_py3.7_cu111
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -873,11 +874,11 @@ workflows: ...@@ -873,11 +874,11 @@ workflows:
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
name: binary_linux_wheel_py3.8_cu112 name: binary_linux_wheel_py3.8_cu111
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -897,11 +898,11 @@ workflows: ...@@ -897,11 +898,11 @@ workflows:
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
name: binary_linux_wheel_py3.9_cu112 name: binary_linux_wheel_py3.9_cu111
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_macos_wheel: - binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -954,13 +955,13 @@ workflows: ...@@ -954,13 +955,13 @@ workflows:
name: binary_win_wheel_py3.6_cu102 name: binary_win_wheel_py3.6_cu102
python_version: '3.6' python_version: '3.6'
- binary_win_wheel: - binary_win_wheel:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: master only: master
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.6'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
...@@ -990,13 +991,13 @@ workflows: ...@@ -990,13 +991,13 @@ workflows:
name: binary_win_wheel_py3.7_cu102 name: binary_win_wheel_py3.7_cu102
python_version: '3.7' python_version: '3.7'
- binary_win_wheel: - binary_win_wheel:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: master only: master
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.7'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
...@@ -1026,13 +1027,13 @@ workflows: ...@@ -1026,13 +1027,13 @@ workflows:
name: binary_win_wheel_py3.8_cu102 name: binary_win_wheel_py3.8_cu102
python_version: '3.8' python_version: '3.8'
- binary_win_wheel: - binary_win_wheel:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: master only: master
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
...@@ -1057,8 +1058,8 @@ workflows: ...@@ -1057,8 +1058,8 @@ workflows:
name: binary_win_wheel_py3.9_cu102 name: binary_win_wheel_py3.9_cu102
python_version: '3.9' python_version: '3.9'
- binary_win_wheel: - binary_win_wheel:
cu_version: cu112 cu_version: cu111
name: binary_win_wheel_py3.9_cu112 name: binary_win_wheel_py3.9_cu111
python_version: '3.9' python_version: '3.9'
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
...@@ -1079,11 +1080,11 @@ workflows: ...@@ -1079,11 +1080,11 @@ workflows:
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
name: binary_linux_conda_py3.6_cu112 name: binary_linux_conda_py3.6_cu111
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1103,11 +1104,11 @@ workflows: ...@@ -1103,11 +1104,11 @@ workflows:
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
name: binary_linux_conda_py3.7_cu112 name: binary_linux_conda_py3.7_cu111
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1127,11 +1128,11 @@ workflows: ...@@ -1127,11 +1128,11 @@ workflows:
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
name: binary_linux_conda_py3.8_cu112 name: binary_linux_conda_py3.8_cu111
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1151,11 +1152,11 @@ workflows: ...@@ -1151,11 +1152,11 @@ workflows:
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
name: binary_linux_conda_py3.9_cu112 name: binary_linux_conda_py3.9_cu111
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_macos_conda: - binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1208,13 +1209,13 @@ workflows: ...@@ -1208,13 +1209,13 @@ workflows:
name: binary_win_conda_py3.6_cu102 name: binary_win_conda_py3.6_cu102
python_version: '3.6' python_version: '3.6'
- binary_win_conda: - binary_win_conda:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: master only: master
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.6'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
...@@ -1244,13 +1245,13 @@ workflows: ...@@ -1244,13 +1245,13 @@ workflows:
name: binary_win_conda_py3.7_cu102 name: binary_win_conda_py3.7_cu102
python_version: '3.7' python_version: '3.7'
- binary_win_conda: - binary_win_conda:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: master only: master
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.7'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
...@@ -1280,13 +1281,13 @@ workflows: ...@@ -1280,13 +1281,13 @@ workflows:
name: binary_win_conda_py3.8_cu102 name: binary_win_conda_py3.8_cu102
python_version: '3.8' python_version: '3.8'
- binary_win_conda: - binary_win_conda:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: master only: master
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
...@@ -1311,8 +1312,8 @@ workflows: ...@@ -1311,8 +1312,8 @@ workflows:
name: binary_win_conda_py3.9_cu102 name: binary_win_conda_py3.9_cu102
python_version: '3.9' python_version: '3.9'
- binary_win_conda: - binary_win_conda:
cu_version: cu112 cu_version: cu111
name: binary_win_conda_py3.9_cu112 name: binary_win_conda_py3.9_cu111
python_version: '3.9' python_version: '3.9'
- build_docs: - build_docs:
name: build_docs name: build_docs
...@@ -1578,16 +1579,16 @@ workflows: ...@@ -1578,16 +1579,16 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.6_cu102_upload - nightly_binary_linux_wheel_py3.6_cu102_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1595,19 +1596,19 @@ workflows: ...@@ -1595,19 +1596,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_wheel_py3.6_cu112 - nightly_binary_linux_wheel_py3.6_cu111
subfolder: cu112/ subfolder: cu111/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.6'
requires: requires:
- nightly_binary_linux_wheel_py3.6_cu112_upload - nightly_binary_linux_wheel_py3.6_cu111_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1702,16 +1703,16 @@ workflows: ...@@ -1702,16 +1703,16 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu102_upload - nightly_binary_linux_wheel_py3.7_cu102_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1719,19 +1720,19 @@ workflows: ...@@ -1719,19 +1720,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_wheel_py3.7_cu112 - nightly_binary_linux_wheel_py3.7_cu111
subfolder: cu112/ subfolder: cu111/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu112_upload - nightly_binary_linux_wheel_py3.7_cu111_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1826,16 +1827,16 @@ workflows: ...@@ -1826,16 +1827,16 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu102_upload - nightly_binary_linux_wheel_py3.8_cu102_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1843,19 +1844,19 @@ workflows: ...@@ -1843,19 +1844,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_wheel_py3.8_cu112 - nightly_binary_linux_wheel_py3.8_cu111
subfolder: cu112/ subfolder: cu111/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu112_upload - nightly_binary_linux_wheel_py3.8_cu111_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -1950,16 +1951,16 @@ workflows: ...@@ -1950,16 +1951,16 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu102_upload - nightly_binary_linux_wheel_py3.9_cu102_upload
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
filters: filters:
...@@ -1967,19 +1968,19 @@ workflows: ...@@ -1967,19 +1968,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_wheel_py3.9_cu112 - nightly_binary_linux_wheel_py3.9_cu111
subfolder: cu112/ subfolder: cu111/
- smoke_test_linux_pip: - smoke_test_linux_pip:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu112_upload - nightly_binary_linux_wheel_py3.9_cu111_upload
- binary_macos_wheel: - binary_macos_wheel:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -2156,13 +2157,13 @@ workflows: ...@@ -2156,13 +2157,13 @@ workflows:
requires: requires:
- nightly_binary_win_wheel_py3.6_cu102_upload - nightly_binary_win_wheel_py3.6_cu102_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.6'
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
...@@ -2171,19 +2172,19 @@ workflows: ...@@ -2171,19 +2172,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_win_wheel_py3.6_cu112 - nightly_binary_win_wheel_py3.6_cu111
subfolder: cu112/ subfolder: cu111/
- smoke_test_win_pip: - smoke_test_win_pip:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.6'
requires: requires:
- nightly_binary_win_wheel_py3.6_cu112_upload - nightly_binary_win_wheel_py3.6_cu111_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2272,13 +2273,13 @@ workflows: ...@@ -2272,13 +2273,13 @@ workflows:
requires: requires:
- nightly_binary_win_wheel_py3.7_cu102_upload - nightly_binary_win_wheel_py3.7_cu102_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.7'
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
...@@ -2287,19 +2288,19 @@ workflows: ...@@ -2287,19 +2288,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_win_wheel_py3.7_cu112 - nightly_binary_win_wheel_py3.7_cu111
subfolder: cu112/ subfolder: cu111/
- smoke_test_win_pip: - smoke_test_win_pip:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.7'
requires: requires:
- nightly_binary_win_wheel_py3.7_cu112_upload - nightly_binary_win_wheel_py3.7_cu111_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2388,13 +2389,13 @@ workflows: ...@@ -2388,13 +2389,13 @@ workflows:
requires: requires:
- nightly_binary_win_wheel_py3.8_cu102_upload - nightly_binary_win_wheel_py3.8_cu102_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
...@@ -2403,19 +2404,19 @@ workflows: ...@@ -2403,19 +2404,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_win_wheel_py3.8_cu112 - nightly_binary_win_wheel_py3.8_cu111
subfolder: cu112/ subfolder: cu111/
- smoke_test_win_pip: - smoke_test_win_pip:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.8'
requires: requires:
- nightly_binary_win_wheel_py3.8_cu112_upload - nightly_binary_win_wheel_py3.8_cu111_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2504,13 +2505,13 @@ workflows: ...@@ -2504,13 +2505,13 @@ workflows:
requires: requires:
- nightly_binary_win_wheel_py3.9_cu102_upload - nightly_binary_win_wheel_py3.9_cu102_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.9'
- binary_wheel_upload: - binary_wheel_upload:
context: org-member context: org-member
...@@ -2519,19 +2520,19 @@ workflows: ...@@ -2519,19 +2520,19 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_win_wheel_py3.9_cu112 - nightly_binary_win_wheel_py3.9_cu111
subfolder: cu112/ subfolder: cu111/
- smoke_test_win_pip: - smoke_test_win_pip:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.9'
requires: requires:
- nightly_binary_win_wheel_py3.9_cu112_upload - nightly_binary_win_wheel_py3.9_cu111_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -2623,16 +2624,16 @@ workflows: ...@@ -2623,16 +2624,16 @@ workflows:
requires: requires:
- nightly_binary_linux_conda_py3.6_cu102_upload - nightly_binary_linux_conda_py3.6_cu102_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
filters: filters:
...@@ -2640,18 +2641,18 @@ workflows: ...@@ -2640,18 +2641,18 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_conda_py3.6_cu112 - nightly_binary_linux_conda_py3.6_cu111
- smoke_test_linux_conda: - smoke_test_linux_conda:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.6'
requires: requires:
- nightly_binary_linux_conda_py3.6_cu112_upload - nightly_binary_linux_conda_py3.6_cu111_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -2743,16 +2744,16 @@ workflows: ...@@ -2743,16 +2744,16 @@ workflows:
requires: requires:
- nightly_binary_linux_conda_py3.7_cu102_upload - nightly_binary_linux_conda_py3.7_cu102_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
filters: filters:
...@@ -2760,18 +2761,18 @@ workflows: ...@@ -2760,18 +2761,18 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_conda_py3.7_cu112 - nightly_binary_linux_conda_py3.7_cu111
- smoke_test_linux_conda: - smoke_test_linux_conda:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.7'
requires: requires:
- nightly_binary_linux_conda_py3.7_cu112_upload - nightly_binary_linux_conda_py3.7_cu111_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -2863,16 +2864,16 @@ workflows: ...@@ -2863,16 +2864,16 @@ workflows:
requires: requires:
- nightly_binary_linux_conda_py3.8_cu102_upload - nightly_binary_linux_conda_py3.8_cu102_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
filters: filters:
...@@ -2880,18 +2881,18 @@ workflows: ...@@ -2880,18 +2881,18 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_conda_py3.8_cu112 - nightly_binary_linux_conda_py3.8_cu111
- smoke_test_linux_conda: - smoke_test_linux_conda:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.8'
requires: requires:
- nightly_binary_linux_conda_py3.8_cu112_upload - nightly_binary_linux_conda_py3.8_cu111_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -2983,16 +2984,16 @@ workflows: ...@@ -2983,16 +2984,16 @@ workflows:
requires: requires:
- nightly_binary_linux_conda_py3.9_cu102_upload - nightly_binary_linux_conda_py3.9_cu102_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda112 conda_docker_image: pytorch/conda-builder:cuda111
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda112 wheel_docker_image: pytorch/manylinux-cuda111
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
filters: filters:
...@@ -3000,18 +3001,18 @@ workflows: ...@@ -3000,18 +3001,18 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_linux_conda_py3.9_cu112 - nightly_binary_linux_conda_py3.9_cu111
- smoke_test_linux_conda: - smoke_test_linux_conda:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.9'
requires: requires:
- nightly_binary_linux_conda_py3.9_cu112_upload - nightly_binary_linux_conda_py3.9_cu111_upload
- binary_macos_conda: - binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cu_version: cpu cu_version: cpu
...@@ -3181,13 +3182,13 @@ workflows: ...@@ -3181,13 +3182,13 @@ workflows:
requires: requires:
- nightly_binary_win_conda_py3.6_cu102_upload - nightly_binary_win_conda_py3.6_cu102_upload
- binary_win_conda: - binary_win_conda:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.6'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -3196,18 +3197,18 @@ workflows: ...@@ -3196,18 +3197,18 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_win_conda_py3.6_cu112 - nightly_binary_win_conda_py3.6_cu111
- smoke_test_win_conda: - smoke_test_win_conda:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.6'
requires: requires:
- nightly_binary_win_conda_py3.6_cu112_upload - nightly_binary_win_conda_py3.6_cu111_upload
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -3293,13 +3294,13 @@ workflows: ...@@ -3293,13 +3294,13 @@ workflows:
requires: requires:
- nightly_binary_win_conda_py3.7_cu102_upload - nightly_binary_win_conda_py3.7_cu102_upload
- binary_win_conda: - binary_win_conda:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.7'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -3308,18 +3309,18 @@ workflows: ...@@ -3308,18 +3309,18 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_win_conda_py3.7_cu112 - nightly_binary_win_conda_py3.7_cu111
- smoke_test_win_conda: - smoke_test_win_conda:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.7'
requires: requires:
- nightly_binary_win_conda_py3.7_cu112_upload - nightly_binary_win_conda_py3.7_cu111_upload
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -3405,13 +3406,13 @@ workflows: ...@@ -3405,13 +3406,13 @@ workflows:
requires: requires:
- nightly_binary_win_conda_py3.8_cu102_upload - nightly_binary_win_conda_py3.8_cu102_upload
- binary_win_conda: - binary_win_conda:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.8'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -3420,18 +3421,18 @@ workflows: ...@@ -3420,18 +3421,18 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_win_conda_py3.8_cu112 - nightly_binary_win_conda_py3.8_cu111
- smoke_test_win_conda: - smoke_test_win_conda:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.8'
requires: requires:
- nightly_binary_win_conda_py3.8_cu112_upload - nightly_binary_win_conda_py3.8_cu111_upload
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -3517,13 +3518,13 @@ workflows: ...@@ -3517,13 +3518,13 @@ workflows:
requires: requires:
- nightly_binary_win_conda_py3.9_cu102_upload - nightly_binary_win_conda_py3.9_cu102_upload
- binary_win_conda: - binary_win_conda:
cu_version: cu112 cu_version: cu111
filters: filters:
branches: branches:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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' python_version: '3.9'
- binary_conda_upload: - binary_conda_upload:
context: org-member context: org-member
...@@ -3532,18 +3533,18 @@ workflows: ...@@ -3532,18 +3533,18 @@ workflows:
only: nightly only: nightly
tags: tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ 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: requires:
- nightly_binary_win_conda_py3.9_cu112 - nightly_binary_win_conda_py3.9_cu111
- smoke_test_win_conda: - smoke_test_win_conda:
filters: filters:
branches: branches:
only: only:
- nightly - 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' python_version: '3.9'
requires: requires:
- nightly_binary_win_conda_py3.9_cu112_upload - nightly_binary_win_conda_py3.9_cu111_upload
docker_build: docker_build:
triggers: triggers:
- schedule: - schedule:
......
.circleci/config.yml.in
View file @ 0e32a12c
...@@ -216,6 +216,7 @@ jobs: ...@@ -216,6 +216,7 @@ jobs:
- designate_upload_channel - designate_upload_channel
- run: - run:
name: Build conda packages name: Build conda packages
no_output_timeout: 20m
command: | command: |
set -ex set -ex
source packaging/windows/internal/vc_install_helper.sh source packaging/windows/internal/vc_install_helper.sh
......
.circleci/regenerate.py
View file @ 0e32a12c
...@@ -20,7 +20,6 @@ import os.path ...@@ -20,7 +20,6 @@ import os.path
PYTHON_VERSIONS = ["3.6", "3.7", "3.8", "3.9"] 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]+/" 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, ...@@ -30,8 +29,8 @@ def build_workflows(prefix='', filter_branch=None, upload=False, indentation=6,
for btype in ["wheel", "conda"]: for btype in ["wheel", "conda"]:
for os_type in ["linux", "macos", "win"]: for os_type in ["linux", "macos", "win"]:
python_versions = PYTHON_VERSIONS python_versions = PYTHON_VERSIONS
cu_versions_dict = {"linux": ["cpu", "cu101", "cu102", "cu112"], cu_versions_dict = {"linux": ["cpu", "cu101", "cu102", "cu111"],
"win": ["cpu", "cu101", "cu102", "cu112"], "win": ["cpu", "cu101", "cu102", "cu111"],
"macos": ["cpu"]} "macos": ["cpu"]}
cu_versions = cu_versions_dict[os_type] cu_versions = cu_versions_dict[os_type]
for python_version in python_versions: for python_version in python_versions:
...@@ -103,6 +102,7 @@ manylinux_images = { ...@@ -103,6 +102,7 @@ manylinux_images = {
"cu101": "pytorch/manylinux-cuda101", "cu101": "pytorch/manylinux-cuda101",
"cu102": "pytorch/manylinux-cuda102", "cu102": "pytorch/manylinux-cuda102",
"cu110": "pytorch/manylinux-cuda110", "cu110": "pytorch/manylinux-cuda110",
"cu111": "pytorch/manylinux-cuda111",
"cu112": "pytorch/manylinux-cuda112", "cu112": "pytorch/manylinux-cuda112",
} }
......
packaging/pkg_helpers.bash
View file @ 0e32a12c
...@@ -60,6 +60,17 @@ setup_cuda() { ...@@ -60,6 +60,17 @@ setup_cuda() {
# https://github.com/pytorch/pytorch/pull/23408 lands # 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" 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) cu110)
if [[ "$OSTYPE" == "msys" ]]; then if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0" export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0"
...@@ -293,6 +304,9 @@ setup_conda_cudatoolkit_constraint() { ...@@ -293,6 +304,9 @@ setup_conda_cudatoolkit_constraint() {
cu112) cu112)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.2,<11.3 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.2,<11.3 # [not osx]"
;; ;;
cu111)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.1,<11.2 # [not osx]"
;;
cu110) cu110)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.0,<11.1 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.0,<11.1 # [not osx]"
;; ;;
...@@ -331,6 +345,9 @@ setup_conda_cudatoolkit_plain_constraint() { ...@@ -331,6 +345,9 @@ setup_conda_cudatoolkit_plain_constraint() {
cu112) cu112)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=11.2" export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=11.2"
;; ;;
cu111)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=11.1"
;;
cu102) cu102)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=10.2" export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=10.2"
;; ;;
......
packaging/torchvision/meta.yaml
View file @ 0e32a12c
...@@ -16,8 +16,7 @@ requirements: ...@@ -16,8 +16,7 @@ requirements:
host: host:
- python - python
- setuptools - setuptools
- numpy >=1.20 # [py>=39] - defaults::numpy >=1.11
- numpy >=1.11 # [py!=39]
{{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT') }} {{ environ.get('CONDA_PYTORCH_BUILD_CONSTRAINT') }}
{{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }} {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
{{ environ.get('CONDA_CPUONLY_FEATURE') }} {{ environ.get('CONDA_CPUONLY_FEATURE') }}
...@@ -28,8 +27,7 @@ requirements: ...@@ -28,8 +27,7 @@ requirements:
- ffmpeg >=4.2 # [not win] - ffmpeg >=4.2 # [not win]
- jpeg <=9b - jpeg <=9b
- pillow >=4.1.1 - pillow >=4.1.1
- numpy >=1.20 # [py>=39] - defaults::numpy >=1.11
- numpy >=1.11 # [py!=39]
{{ environ.get('CONDA_PYTORCH_CONSTRAINT') }} {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
{{ environ.get('CONDA_CUDATOOLKIT_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 ...@@ -19,6 +19,7 @@ if %CUDA_VER% EQU 100 goto cuda100
if %CUDA_VER% EQU 101 goto cuda101 if %CUDA_VER% EQU 101 goto cuda101
if %CUDA_VER% EQU 102 goto cuda102 if %CUDA_VER% EQU 102 goto cuda102
if %CUDA_VER% EQU 110 goto cuda110 if %CUDA_VER% EQU 110 goto cuda110
if %CUDA_VER% EQU 111 goto cuda111
if %CUDA_VER% EQU 112 goto cuda112 if %CUDA_VER% EQU 112 goto cuda112
echo CUDA %CUDA_VERSION_STR% is not supported 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" ( ...@@ -108,6 +109,27 @@ if not exist "%SRC_DIR%\temp_build\cudnn-11.0-windows-x64-v8.0.4.30.zip" (
goto cuda_common 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 :cuda112
if not exist "%SRC_DIR%\temp_build\cuda_11.2.0_460.89_win10.exe" ( 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