You need to sign in or sign up before continuing.
Unverified Commit 3d9d2641 authored by Nikita Shulga's avatar Nikita Shulga Committed by GitHub
Browse files

Add CUDA-11.3 builds to torchvision (#4248)

* Add CUDA-11.3 builds to torchvision

Also mark  `.cicrleci/config.yml` as autogenerated in `.gitattributes`

* Add CUDA-11.3 install steps for Windows

* And here as well
parent 3fa2055b
...@@ -1008,6 +1008,12 @@ workflows: ...@@ -1008,6 +1008,12 @@ workflows:
name: binary_linux_wheel_py3.6_cu111 name: binary_linux_wheel_py3.6_cu111
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
name: binary_linux_wheel_py3.6_cu113
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.1
name: binary_linux_wheel_py3.6_rocm4.1 name: binary_linux_wheel_py3.6_rocm4.1
...@@ -1041,6 +1047,12 @@ workflows: ...@@ -1041,6 +1047,12 @@ workflows:
name: binary_linux_wheel_py3.7_cu111 name: binary_linux_wheel_py3.7_cu111
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
name: binary_linux_wheel_py3.7_cu113
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.1
name: binary_linux_wheel_py3.7_rocm4.1 name: binary_linux_wheel_py3.7_rocm4.1
...@@ -1069,6 +1081,12 @@ workflows: ...@@ -1069,6 +1081,12 @@ workflows:
name: binary_linux_wheel_py3.8_cu111 name: binary_linux_wheel_py3.8_cu111
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
name: binary_linux_wheel_py3.8_cu113
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.1
name: binary_linux_wheel_py3.8_rocm4.1 name: binary_linux_wheel_py3.8_rocm4.1
...@@ -1097,6 +1115,12 @@ workflows: ...@@ -1097,6 +1115,12 @@ workflows:
name: binary_linux_wheel_py3.9_cu111 name: binary_linux_wheel_py3.9_cu111
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
name: binary_linux_wheel_py3.9_cu113
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.1
name: binary_linux_wheel_py3.9_rocm4.1 name: binary_linux_wheel_py3.9_rocm4.1
...@@ -1158,6 +1182,15 @@ workflows: ...@@ -1158,6 +1182,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.6_cu111 name: binary_win_wheel_py3.6_cu111
python_version: '3.6' python_version: '3.6'
- binary_win_wheel:
cu_version: cu113
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.6_cu113
python_version: '3.6'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1185,6 +1218,15 @@ workflows: ...@@ -1185,6 +1218,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.7_cu111 name: binary_win_wheel_py3.7_cu111
python_version: '3.7' python_version: '3.7'
- binary_win_wheel:
cu_version: cu113
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.7_cu113
python_version: '3.7'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1212,6 +1254,15 @@ workflows: ...@@ -1212,6 +1254,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.8_cu111 name: binary_win_wheel_py3.8_cu111
python_version: '3.8' python_version: '3.8'
- binary_win_wheel:
cu_version: cu113
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.8_cu113
python_version: '3.8'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
name: binary_win_wheel_py3.9_cpu name: binary_win_wheel_py3.9_cpu
...@@ -1227,8 +1278,17 @@ workflows: ...@@ -1227,8 +1278,17 @@ workflows:
python_version: '3.9' python_version: '3.9'
- binary_win_wheel: - binary_win_wheel:
cu_version: cu111 cu_version: cu111
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.9_cu111 name: binary_win_wheel_py3.9_cu111
python_version: '3.9' python_version: '3.9'
- binary_win_wheel:
cu_version: cu113
name: binary_win_wheel_py3.9_cu113
python_version: '3.9'
- 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
...@@ -1247,6 +1307,12 @@ workflows: ...@@ -1247,6 +1307,12 @@ workflows:
name: binary_linux_conda_py3.6_cu111 name: binary_linux_conda_py3.6_cu111
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
name: binary_linux_conda_py3.6_cu113
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda113
- 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
...@@ -1265,6 +1331,12 @@ workflows: ...@@ -1265,6 +1331,12 @@ workflows:
name: binary_linux_conda_py3.7_cu111 name: binary_linux_conda_py3.7_cu111
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
name: binary_linux_conda_py3.7_cu113
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda113
- 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
...@@ -1283,6 +1355,12 @@ workflows: ...@@ -1283,6 +1355,12 @@ workflows:
name: binary_linux_conda_py3.8_cu111 name: binary_linux_conda_py3.8_cu111
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
name: binary_linux_conda_py3.8_cu113
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda113
- 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
...@@ -1301,6 +1379,12 @@ workflows: ...@@ -1301,6 +1379,12 @@ workflows:
name: binary_linux_conda_py3.9_cu111 name: binary_linux_conda_py3.9_cu111
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda111 wheel_docker_image: pytorch/manylinux-cuda111
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
name: binary_linux_conda_py3.9_cu113
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda113
- 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
...@@ -1352,6 +1436,15 @@ workflows: ...@@ -1352,6 +1436,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.6_cu111 name: binary_win_conda_py3.6_cu111
python_version: '3.6' python_version: '3.6'
- binary_win_conda:
cu_version: cu113
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.6_cu113
python_version: '3.6'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1379,6 +1472,15 @@ workflows: ...@@ -1379,6 +1472,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.7_cu111 name: binary_win_conda_py3.7_cu111
python_version: '3.7' python_version: '3.7'
- binary_win_conda:
cu_version: cu113
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.7_cu113
python_version: '3.7'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1406,6 +1508,15 @@ workflows: ...@@ -1406,6 +1508,15 @@ workflows:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/ only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.8_cu111 name: binary_win_conda_py3.8_cu111
python_version: '3.8' python_version: '3.8'
- binary_win_conda:
cu_version: cu113
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.8_cu113
python_version: '3.8'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
name: binary_win_conda_py3.9_cpu name: binary_win_conda_py3.9_cpu
...@@ -1421,8 +1532,17 @@ workflows: ...@@ -1421,8 +1532,17 @@ workflows:
python_version: '3.9' python_version: '3.9'
- binary_win_conda: - binary_win_conda:
cu_version: cu111 cu_version: cu111
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.9_cu111 name: binary_win_conda_py3.9_cu111
python_version: '3.9' python_version: '3.9'
- binary_win_conda:
cu_version: cu113
name: binary_win_conda_py3.9_cu113
python_version: '3.9'
- build_docs: - build_docs:
filters: filters:
branches: branches:
...@@ -1741,6 +1861,37 @@ workflows: ...@@ -1741,6 +1861,37 @@ workflows:
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_linux_wheel_py3.6_cu111_upload - nightly_binary_linux_wheel_py3.6_cu111_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_cu113
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_cu113_upload
requires:
- nightly_binary_linux_wheel_py3.6_cu113
subfolder: cu113/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.6_cu113_smoke_test_pip
python_version: '3.6'
requires:
- nightly_binary_linux_wheel_py3.6_cu113_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.1
filters: filters:
...@@ -1894,6 +2045,37 @@ workflows: ...@@ -1894,6 +2045,37 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu111_upload - nightly_binary_linux_wheel_py3.7_cu111_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu113
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu113_upload
requires:
- nightly_binary_linux_wheel_py3.7_cu113
subfolder: cu113/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.7_cu113_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_cu113_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.1
filters: filters:
...@@ -2047,6 +2229,37 @@ workflows: ...@@ -2047,6 +2229,37 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu111_upload - nightly_binary_linux_wheel_py3.8_cu111_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu113
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu113_upload
requires:
- nightly_binary_linux_wheel_py3.8_cu113
subfolder: cu113/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.8_cu113_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_cu113_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.1
filters: filters:
...@@ -2200,6 +2413,37 @@ workflows: ...@@ -2200,6 +2413,37 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu111_upload - nightly_binary_linux_wheel_py3.9_cu111_upload
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu113
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu113_upload
requires:
- nightly_binary_linux_wheel_py3.9_cu113
subfolder: cu113/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.9_cu113_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_cu113_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: rocm4.1 cu_version: rocm4.1
filters: filters:
...@@ -2435,6 +2679,35 @@ workflows: ...@@ -2435,6 +2679,35 @@ workflows:
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_win_wheel_py3.6_cu111_upload - nightly_binary_win_wheel_py3.6_cu111_upload
- binary_win_wheel:
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.6_cu113
python_version: '3.6'
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.6_cu113_upload
requires:
- nightly_binary_win_wheel_py3.6_cu113
subfolder: cu113/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.6_cu113_smoke_test_pip
python_version: '3.6'
requires:
- nightly_binary_win_wheel_py3.6_cu113_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2522,6 +2795,35 @@ workflows: ...@@ -2522,6 +2795,35 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_win_wheel_py3.7_cu111_upload - nightly_binary_win_wheel_py3.7_cu111_upload
- binary_win_wheel:
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.7_cu113
python_version: '3.7'
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.7_cu113_upload
requires:
- nightly_binary_win_wheel_py3.7_cu113
subfolder: cu113/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.7_cu113_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_win_wheel_py3.7_cu113_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2609,6 +2911,35 @@ workflows: ...@@ -2609,6 +2911,35 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_win_wheel_py3.8_cu111_upload - nightly_binary_win_wheel_py3.8_cu111_upload
- binary_win_wheel:
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu113
python_version: '3.8'
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu113_upload
requires:
- nightly_binary_win_wheel_py3.8_cu113
subfolder: cu113/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.8_cu113_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_win_wheel_py3.8_cu113_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2696,6 +3027,35 @@ workflows: ...@@ -2696,6 +3027,35 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_win_wheel_py3.9_cu111_upload - nightly_binary_win_wheel_py3.9_cu111_upload
- binary_win_wheel:
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.9_cu113
python_version: '3.9'
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.9_cu113_upload
requires:
- nightly_binary_win_wheel_py3.9_cu113
subfolder: cu113/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.9_cu113_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_win_wheel_py3.9_cu113_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
...@@ -2786,6 +3146,36 @@ workflows: ...@@ -2786,6 +3146,36 @@ workflows:
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_linux_conda_py3.6_cu111_upload - nightly_binary_linux_conda_py3.6_cu111_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_cu113
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_cu113_upload
requires:
- nightly_binary_linux_conda_py3.6_cu113
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.6_cu113_smoke_test_conda
python_version: '3.6'
requires:
- nightly_binary_linux_conda_py3.6_cu113_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
...@@ -2876,6 +3266,36 @@ workflows: ...@@ -2876,6 +3266,36 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_conda_py3.7_cu111_upload - nightly_binary_linux_conda_py3.7_cu111_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu113
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu113_upload
requires:
- nightly_binary_linux_conda_py3.7_cu113
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.7_cu113_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_linux_conda_py3.7_cu113_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
...@@ -2966,6 +3386,36 @@ workflows: ...@@ -2966,6 +3386,36 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_conda_py3.8_cu111_upload - nightly_binary_linux_conda_py3.8_cu111_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu113
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu113_upload
requires:
- nightly_binary_linux_conda_py3.8_cu113
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.8_cu113_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_linux_conda_py3.8_cu113_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
...@@ -3056,6 +3506,36 @@ workflows: ...@@ -3056,6 +3506,36 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_conda_py3.9_cu111_upload - nightly_binary_linux_conda_py3.9_cu111_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda113
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu113
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda113
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu113_upload
requires:
- nightly_binary_linux_conda_py3.9_cu113
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.9_cu113_smoke_test_conda
python_version: '3.9'
requires:
- nightly_binary_linux_conda_py3.9_cu113_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
...@@ -3224,6 +3704,34 @@ workflows: ...@@ -3224,6 +3704,34 @@ workflows:
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_win_conda_py3.6_cu111_upload - nightly_binary_win_conda_py3.6_cu111_upload
- binary_win_conda:
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.6_cu113
python_version: '3.6'
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.6_cu113_upload
requires:
- nightly_binary_win_conda_py3.6_cu113
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.6_cu113_smoke_test_conda
python_version: '3.6'
requires:
- nightly_binary_win_conda_py3.6_cu113_upload
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -3308,6 +3816,34 @@ workflows: ...@@ -3308,6 +3816,34 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_win_conda_py3.7_cu111_upload - nightly_binary_win_conda_py3.7_cu111_upload
- binary_win_conda:
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.7_cu113
python_version: '3.7'
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.7_cu113_upload
requires:
- nightly_binary_win_conda_py3.7_cu113
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.7_cu113_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_win_conda_py3.7_cu113_upload
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -3392,6 +3928,34 @@ workflows: ...@@ -3392,6 +3928,34 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_win_conda_py3.8_cu111_upload - nightly_binary_win_conda_py3.8_cu111_upload
- binary_win_conda:
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu113
python_version: '3.8'
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu113_upload
requires:
- nightly_binary_win_conda_py3.8_cu113
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.8_cu113_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_win_conda_py3.8_cu113_upload
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -3476,6 +4040,34 @@ workflows: ...@@ -3476,6 +4040,34 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_win_conda_py3.9_cu111_upload - nightly_binary_win_conda_py3.9_cu111_upload
- binary_win_conda:
cu_version: cu113
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.9_cu113
python_version: '3.9'
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.9_cu113_upload
requires:
- nightly_binary_win_conda_py3.9_cu113
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.9_cu113_smoke_test_conda
python_version: '3.9'
requires:
- nightly_binary_win_conda_py3.9_cu113_upload
docker_build: docker_build:
triggers: triggers:
- schedule: - schedule:
......
...@@ -30,8 +30,8 @@ def build_workflows(prefix='', filter_branch=None, upload=False, indentation=6, ...@@ -30,8 +30,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", "cu102", "cu111", "rocm4.1", "rocm4.2"], cu_versions_dict = {"linux": ["cpu", "cu102", "cu111", "cu113", "rocm4.1", "rocm4.2"],
"win": ["cpu", "cu102", "cu111"], "win": ["cpu", "cu102", "cu111", "cu113"],
"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:
...@@ -115,6 +115,7 @@ manylinux_images = { ...@@ -115,6 +115,7 @@ manylinux_images = {
"cu110": "pytorch/manylinux-cuda110", "cu110": "pytorch/manylinux-cuda110",
"cu111": "pytorch/manylinux-cuda111", "cu111": "pytorch/manylinux-cuda111",
"cu112": "pytorch/manylinux-cuda112", "cu112": "pytorch/manylinux-cuda112",
"cu113": "pytorch/manylinux-cuda113",
} }
......
...@@ -6,3 +6,6 @@ ...@@ -6,3 +6,6 @@
# To ignore it use below # To ignore it use below
*.ipynb linguist-documentation *.ipynb linguist-documentation
# To exclude autogenerated files from code reviews
.circleci/config.yml linguist-generated=true
...@@ -46,6 +46,15 @@ setup_cuda() { ...@@ -46,6 +46,15 @@ setup_cuda() {
# Now work out the CUDA settings # Now work out the CUDA settings
case "$CU_VERSION" in case "$CU_VERSION" in
cu113)
if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.3"
else
export CUDA_HOME=/usr/local/cuda-11.3/
fi
export FORCE_CUDA=1
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
;;
cu112) cu112)
if [[ "$OSTYPE" == "msys" ]]; then if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.2" export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.2"
...@@ -283,6 +292,9 @@ setup_conda_cudatoolkit_constraint() { ...@@ -283,6 +292,9 @@ setup_conda_cudatoolkit_constraint() {
export CONDA_CUDATOOLKIT_CONSTRAINT="" export CONDA_CUDATOOLKIT_CONSTRAINT=""
else else
case "$CU_VERSION" in case "$CU_VERSION" in
cu113)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.3,<11.4 # [not osx]"
;;
cu112) cu112)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.2,<11.3 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.2,<11.3 # [not osx]"
;; ;;
...@@ -324,6 +336,9 @@ setup_conda_cudatoolkit_plain_constraint() { ...@@ -324,6 +336,9 @@ setup_conda_cudatoolkit_plain_constraint() {
export CMAKE_USE_CUDA=0 export CMAKE_USE_CUDA=0
else else
case "$CU_VERSION" in case "$CU_VERSION" in
cu113)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=11.3"
;;
cu112) cu112)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=11.2" export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=11.2"
;; ;;
......
...@@ -21,6 +21,7 @@ if %CUDA_VER% EQU 102 goto cuda102 ...@@ -21,6 +21,7 @@ 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 111 goto cuda111
if %CUDA_VER% EQU 112 goto cuda112 if %CUDA_VER% EQU 112 goto cuda112
if %CUDA_VER% EQU 113 goto cuda113
echo CUDA %CUDA_VERSION_STR% is not supported echo CUDA %CUDA_VERSION_STR% is not supported
exit /b 1 exit /b 1
...@@ -147,6 +148,26 @@ if not exist "%SRC_DIR%\temp_build\cudnn-11.2-windows-x64-v8.1.0.77.zip" ( ...@@ -147,6 +148,26 @@ if not exist "%SRC_DIR%\temp_build\cudnn-11.2-windows-x64-v8.1.0.77.zip" (
goto cuda_common goto cuda_common
:cuda113
set CUDA_INSTALL_EXE=cuda_11.3.0_465.89_win10.exe
if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" (
curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
set "ARGS=thrust_11.3 nvcc_11.3 cuobjdump_11.3 nvprune_11.3 nvprof_11.3 cupti_11.3 cublas_11.3 cublas_dev_11.3 cudart_11.3 cufft_11.3 cufft_dev_11.3 curand_11.3 curand_dev_11.3 cusolver_11.3 cusolver_dev_11.3 cusparse_11.3 cusparse_dev_11.3 npp_11.3 npp_dev_11.3 nvrtc_11.3 nvrtc_dev_11.3 nvml_dev_11.3"
)
set CUDNN_INSTALL_ZIP=cudnn-11.3-windows-x64-v8.2.0.53.zip
if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" (
curl -k -L "http://s3.amazonaws.com/ossci-windows/%CUDNN_INSTALL_ZIP%" --output "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
)
goto cuda_common
:cuda_common :cuda_common
if not exist "%SRC_DIR%\temp_build\NvToolsExt.7z" ( if not exist "%SRC_DIR%\temp_build\NvToolsExt.7z" (
......
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