Skip to content

GitLab

Unverified Commit ca26022c authored by peterjc123's avatar peterjc123 Committed by GitHub
Browse files

Add CUDA 11 binary builds (#2671)

parent 38aab83a
Hide whitespace changes
Inline Side-by-side
Showing with 468 additions and 30 deletions
.circleci/config.yml
View file @ ca26022c
...@@ -645,6 +645,11 @@ workflows: ...@@ -645,6 +645,11 @@ workflows:
name: binary_linux_wheel_py3.6_cu102 name: binary_linux_wheel_py3.6_cu102
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
cu_version: cu110
name: binary_linux_wheel_py3.6_cu110
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda110
- binary_linux_wheel: - binary_linux_wheel:
cu_version: cpu cu_version: cpu
name: binary_linux_wheel_py3.7_cpu name: binary_linux_wheel_py3.7_cpu
...@@ -665,6 +670,11 @@ workflows: ...@@ -665,6 +670,11 @@ workflows:
name: binary_linux_wheel_py3.7_cu102 name: binary_linux_wheel_py3.7_cu102
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
cu_version: cu110
name: binary_linux_wheel_py3.7_cu110
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda110
- binary_linux_wheel: - binary_linux_wheel:
cu_version: cpu cu_version: cpu
name: binary_linux_wheel_py3.8_cpu name: binary_linux_wheel_py3.8_cpu
...@@ -685,6 +695,11 @@ workflows: ...@@ -685,6 +695,11 @@ workflows:
name: binary_linux_wheel_py3.8_cu102 name: binary_linux_wheel_py3.8_cu102
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
cu_version: cu110
name: binary_linux_wheel_py3.8_cu110
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda110
- binary_macos_wheel: - binary_macos_wheel:
cu_version: cpu cu_version: cpu
name: binary_macos_wheel_py3.6_cpu name: binary_macos_wheel_py3.6_cpu
...@@ -727,6 +742,15 @@ workflows: ...@@ -727,6 +742,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_cu102 name: binary_win_wheel_py3.6_cu102
python_version: '3.6' python_version: '3.6'
- binary_win_wheel:
cu_version: cu110
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.6_cu110
python_version: '3.6'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -754,6 +778,15 @@ workflows: ...@@ -754,6 +778,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_cu102 name: binary_win_wheel_py3.7_cu102
python_version: '3.7' python_version: '3.7'
- binary_win_wheel:
cu_version: cu110
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_wheel_py3.7_cu110
python_version: '3.7'
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
name: binary_win_wheel_py3.8_cpu name: binary_win_wheel_py3.8_cpu
...@@ -769,8 +802,17 @@ workflows: ...@@ -769,8 +802,17 @@ workflows:
python_version: '3.8' python_version: '3.8'
- binary_win_wheel: - binary_win_wheel:
cu_version: cu102 cu_version: cu102
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
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:
cu_version: cu110
name: binary_win_wheel_py3.8_cu110
python_version: '3.8'
- binary_linux_conda: - binary_linux_conda:
cu_version: cpu cu_version: cpu
name: binary_linux_conda_py3.6_cpu name: binary_linux_conda_py3.6_cpu
...@@ -791,6 +833,11 @@ workflows: ...@@ -791,6 +833,11 @@ workflows:
name: binary_linux_conda_py3.6_cu102 name: binary_linux_conda_py3.6_cu102
python_version: '3.6' python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda:
cu_version: cu110
name: binary_linux_conda_py3.6_cu110
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda110
- binary_linux_conda: - binary_linux_conda:
cu_version: cpu cu_version: cpu
name: binary_linux_conda_py3.7_cpu name: binary_linux_conda_py3.7_cpu
...@@ -811,6 +858,11 @@ workflows: ...@@ -811,6 +858,11 @@ workflows:
name: binary_linux_conda_py3.7_cu102 name: binary_linux_conda_py3.7_cu102
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda:
cu_version: cu110
name: binary_linux_conda_py3.7_cu110
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda110
- binary_linux_conda: - binary_linux_conda:
cu_version: cpu cu_version: cpu
name: binary_linux_conda_py3.8_cpu name: binary_linux_conda_py3.8_cpu
...@@ -831,6 +883,11 @@ workflows: ...@@ -831,6 +883,11 @@ workflows:
name: binary_linux_conda_py3.8_cu102 name: binary_linux_conda_py3.8_cu102
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda:
cu_version: cu110
name: binary_linux_conda_py3.8_cu110
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda110
- binary_macos_conda: - binary_macos_conda:
cu_version: cpu cu_version: cpu
name: binary_macos_conda_py3.6_cpu name: binary_macos_conda_py3.6_cpu
...@@ -873,6 +930,15 @@ workflows: ...@@ -873,6 +930,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_cu102 name: binary_win_conda_py3.6_cu102
python_version: '3.6' python_version: '3.6'
- binary_win_conda:
cu_version: cu110
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.6_cu110
python_version: '3.6'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -900,6 +966,15 @@ workflows: ...@@ -900,6 +966,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_cu102 name: binary_win_conda_py3.7_cu102
python_version: '3.7' python_version: '3.7'
- binary_win_conda:
cu_version: cu110
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: binary_win_conda_py3.7_cu110
python_version: '3.7'
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
name: binary_win_conda_py3.8_cpu name: binary_win_conda_py3.8_cpu
...@@ -915,8 +990,17 @@ workflows: ...@@ -915,8 +990,17 @@ workflows:
python_version: '3.8' python_version: '3.8'
- binary_win_conda: - binary_win_conda:
cu_version: cu102 cu_version: cu102
filters:
branches:
only: master
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
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:
cu_version: cu110
name: binary_win_conda_py3.8_cu110
python_version: '3.8'
- python_lint - python_lint
- python_type_check - python_type_check
- clang_format - clang_format
...@@ -1138,6 +1222,36 @@ workflows: ...@@ -1138,6 +1222,36 @@ workflows:
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_linux_wheel_py3.6_cu102_upload - nightly_binary_linux_wheel_py3.6_cu102_upload
- binary_linux_wheel:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.6_cu110
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda110
- 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_cu110_upload
requires:
- nightly_binary_linux_wheel_py3.6_cu110
subfolder: cu110/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.6_cu110_smoke_test_pip
python_version: '3.6'
requires:
- nightly_binary_linux_wheel_py3.6_cu110_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1258,6 +1372,36 @@ workflows: ...@@ -1258,6 +1372,36 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu102_upload - nightly_binary_linux_wheel_py3.7_cu102_upload
- binary_linux_wheel:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu110
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda110
- 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_cu110_upload
requires:
- nightly_binary_linux_wheel_py3.7_cu110
subfolder: cu110/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.7_cu110_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_cu110_upload
- binary_linux_wheel: - binary_linux_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1378,6 +1522,36 @@ workflows: ...@@ -1378,6 +1522,36 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu102_upload - nightly_binary_linux_wheel_py3.8_cu102_upload
- binary_linux_wheel:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu110
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda110
- 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_cu110_upload
requires:
- nightly_binary_linux_wheel_py3.8_cu110
subfolder: cu110/
- smoke_test_linux_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_wheel_py3.8_cu110_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_cu110_upload
- binary_macos_wheel: - binary_macos_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1528,6 +1702,35 @@ workflows: ...@@ -1528,6 +1702,35 @@ workflows:
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_win_wheel_py3.6_cu102_upload - nightly_binary_win_wheel_py3.6_cu102_upload
- binary_win_wheel:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.6_cu110
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_cu110_upload
requires:
- nightly_binary_win_wheel_py3.6_cu110
subfolder: cu110/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.6_cu110_smoke_test_pip
python_version: '3.6'
requires:
- nightly_binary_win_wheel_py3.6_cu110_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1615,6 +1818,35 @@ workflows: ...@@ -1615,6 +1818,35 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_win_wheel_py3.7_cu102_upload - nightly_binary_win_wheel_py3.7_cu102_upload
- binary_win_wheel:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.7_cu110
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_cu110_upload
requires:
- nightly_binary_win_wheel_py3.7_cu110
subfolder: cu110/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.7_cu110_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_win_wheel_py3.7_cu110_upload
- binary_win_wheel: - binary_win_wheel:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1702,6 +1934,35 @@ workflows: ...@@ -1702,6 +1934,35 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_win_wheel_py3.8_cu102_upload - nightly_binary_win_wheel_py3.8_cu102_upload
- binary_win_wheel:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_wheel_py3.8_cu110
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_cu110_upload
requires:
- nightly_binary_win_wheel_py3.8_cu110
subfolder: cu110/
- smoke_test_win_pip:
filters:
branches:
only:
- nightly
name: nightly_binary_win_wheel_py3.8_cu110_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_win_wheel_py3.8_cu110_upload
- binary_linux_conda: - binary_linux_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1818,6 +2079,35 @@ workflows: ...@@ -1818,6 +2079,35 @@ workflows:
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_linux_conda_py3.6_cu102_upload - nightly_binary_linux_conda_py3.6_cu102_upload
- binary_linux_conda:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.6_cu110
python_version: '3.6'
wheel_docker_image: pytorch/manylinux-cuda110
- 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_cu110_upload
requires:
- nightly_binary_linux_conda_py3.6_cu110
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.6_cu110_smoke_test_conda
python_version: '3.6'
requires:
- nightly_binary_linux_conda_py3.6_cu110_upload
- binary_linux_conda: - binary_linux_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -1934,6 +2224,35 @@ workflows: ...@@ -1934,6 +2224,35 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_conda_py3.7_cu102_upload - nightly_binary_linux_conda_py3.7_cu102_upload
- binary_linux_conda:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu110
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda110
- 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_cu110_upload
requires:
- nightly_binary_linux_conda_py3.7_cu110
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.7_cu110_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_linux_conda_py3.7_cu110_upload
- binary_linux_conda: - binary_linux_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2050,6 +2369,35 @@ workflows: ...@@ -2050,6 +2369,35 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_conda_py3.8_cu102_upload - nightly_binary_linux_conda_py3.8_cu102_upload
- binary_linux_conda:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu110
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda110
- 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_cu110_upload
requires:
- nightly_binary_linux_conda_py3.8_cu110
- smoke_test_linux_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_linux_conda_py3.8_cu110_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_linux_conda_py3.8_cu110_upload
- binary_macos_conda: - binary_macos_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2194,6 +2542,34 @@ workflows: ...@@ -2194,6 +2542,34 @@ workflows:
python_version: '3.6' python_version: '3.6'
requires: requires:
- nightly_binary_win_conda_py3.6_cu102_upload - nightly_binary_win_conda_py3.6_cu102_upload
- binary_win_conda:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.6_cu110
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_cu110_upload
requires:
- nightly_binary_win_conda_py3.6_cu110
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.6_cu110_smoke_test_conda
python_version: '3.6'
requires:
- nightly_binary_win_conda_py3.6_cu110_upload
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2278,6 +2654,34 @@ workflows: ...@@ -2278,6 +2654,34 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_win_conda_py3.7_cu102_upload - nightly_binary_win_conda_py3.7_cu102_upload
- binary_win_conda:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.7_cu110
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_cu110_upload
requires:
- nightly_binary_win_conda_py3.7_cu110
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.7_cu110_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_win_conda_py3.7_cu110_upload
- binary_win_conda: - binary_win_conda:
cu_version: cpu cu_version: cpu
filters: filters:
...@@ -2362,6 +2766,34 @@ workflows: ...@@ -2362,6 +2766,34 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_win_conda_py3.8_cu102_upload - nightly_binary_win_conda_py3.8_cu102_upload
- binary_win_conda:
cu_version: cu110
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_win_conda_py3.8_cu110
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_cu110_upload
requires:
- nightly_binary_win_conda_py3.8_cu110
- smoke_test_win_conda:
filters:
branches:
only:
- nightly
name: nightly_binary_win_conda_py3.8_cu110_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_win_conda_py3.8_cu110_upload
docker_build: docker_build:
triggers: triggers:
- schedule: - schedule:
......
.circleci/regenerate.py
View file @ ca26022c
...@@ -27,8 +27,8 @@ def build_workflows(prefix='', filter_branch=None, upload=False, indentation=6, ...@@ -27,8 +27,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", "cu92", "cu101", "cu102"], cu_versions_dict = {"linux": ["cpu", "cu92", "cu101", "cu102", "cu110"],
"win": ["cpu", "cu101", "cu102"], "win": ["cpu", "cu101", "cu102", "cu110"],
"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:
...@@ -69,6 +69,7 @@ manylinux_images = { ...@@ -69,6 +69,7 @@ manylinux_images = {
"cu92": "pytorch/manylinux-cuda92", "cu92": "pytorch/manylinux-cuda92",
"cu101": "pytorch/manylinux-cuda101", "cu101": "pytorch/manylinux-cuda101",
"cu102": "pytorch/manylinux-cuda102", "cu102": "pytorch/manylinux-cuda102",
"cu110": "pytorch/manylinux-cuda110",
} }
......
packaging/pkg_helpers.bash
View file @ ca26022c
...@@ -49,6 +49,17 @@ setup_cuda() { ...@@ -49,6 +49,17 @@ setup_cuda() {
# Now work out the CUDA settings # Now work out the CUDA settings
case "$CU_VERSION" in case "$CU_VERSION" in
cu110)
if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0"
else
export CUDA_HOME=/usr/local/cuda-11.0/
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_50,code=compute_50"
;;
cu102) cu102)
if [[ "$OSTYPE" == "msys" ]]; then if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2" export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2"
...@@ -265,6 +276,9 @@ setup_conda_cudatoolkit_constraint() { ...@@ -265,6 +276,9 @@ setup_conda_cudatoolkit_constraint() {
export CONDA_CUDATOOLKIT_CONSTRAINT="" export CONDA_CUDATOOLKIT_CONSTRAINT=""
else else
case "$CU_VERSION" in case "$CU_VERSION" in
cu110)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.0,<11.1 # [not osx]"
;;
cu102) cu102)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=10.2,<10.3 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=10.2,<10.3 # [not osx]"
;; ;;
......
packaging/torchvision/bld.bat deleted 100644 → 0
View file @ 38aab83a
@echo on
set TORCHVISION_BUILD_VERSION=%PKG_VERSION%
set TORCHVISION_BUILD_NUMBER=%PKG_BUILDNUM%
set build_with_cuda=
if "%CUDA_VERSION%" == "None" goto cuda_flags_end
if "%CUDA_VERSION%" == "cpu" goto cuda_flags_end
if "%CUDA_VERSION%" == "" goto cuda_flags_end
set build_with_cuda=1
set desired_cuda=%CUDA_VERSION:~0,-1%.%CUDA_VERSION:~-1,1%
set CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v%desired_cuda%
set CUDA_BIN_PATH=%CUDA_PATH%\bin
set NVCC_FLAGS=-D__CUDA_NO_HALF_OPERATORS__ --expt-relaxed-constexpr
if "%desired_cuda%" == "9.0" set NVCC_FLAGS=%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_50,code=compute_50
if "%desired_cuda%" == "9.2" set NVCC_FLAGS=%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_61,code=sm_61 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_50,code=compute_50
if "%desired_cuda%" == "10.0" set NVCC_FLAGS=%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_50,code=compute_50
if "%desired_cuda%" == "10.1" set NVCC_FLAGS=%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_50,code=compute_50
if "%desired_cuda%" == "10.2" set NVCC_FLAGS=%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_50,code=compute_50
:cuda_flags_end
python setup.py install --single-version-externally-managed --record=record.txt
if errorlevel 1 exit /b 1
packaging/torchvision/meta.yaml
View file @ ca26022c
...@@ -29,7 +29,7 @@ requirements: ...@@ -29,7 +29,7 @@ requirements:
build: build:
string: py{{py}}_{{ environ['CU_VERSION'] }} string: py{{py}}_{{ environ['CU_VERSION'] }}
script: python setup.py install --single-version-externally-managed --record=record.txt # [not win] script: python setup.py install --single-version-externally-managed --record=record.txt
script_env: script_env:
- CUDA_HOME - CUDA_HOME
- FORCE_CUDA - FORCE_CUDA
......
packaging/windows/internal/cuda_install.bat
View file @ ca26022c
...@@ -18,6 +18,7 @@ if %CUDA_VER% EQU 92 goto cuda92 ...@@ -18,6 +18,7 @@ if %CUDA_VER% EQU 92 goto cuda92
if %CUDA_VER% EQU 100 goto cuda100 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
echo CUDA %CUDA_VERSION_STR% is not supported echo CUDA %CUDA_VERSION_STR% is not supported
exit /b 1 exit /b 1
...@@ -89,6 +90,23 @@ if not exist "%SRC_DIR%\temp_build\cudnn-10.2-windows10-x64-v7.6.5.32.zip" ( ...@@ -89,6 +90,23 @@ if not exist "%SRC_DIR%\temp_build\cudnn-10.2-windows10-x64-v7.6.5.32.zip" (
goto cuda_common goto cuda_common
:cuda110
if not exist "%SRC_DIR%\temp_build\cuda_11.0.2_451.48_win10.exe" (
curl -k -L https://ossci-windows.s3.amazonaws.com/cuda_11.0.2_451.48_win10.exe --output "%SRC_DIR%\temp_build\cuda_11.0.2_451.48_win10.exe"
if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\cuda_11.0.2_451.48_win10.exe"
set "ARGS=nvcc_11.0 cuobjdump_11.0 nvprune_11.0 nvprof_11.0 cupti_11.0 cublas_11.0 cublas_dev_11.0 cudart_11.0 cufft_11.0 cufft_dev_11.0 curand_11.0 curand_dev_11.0 cusolver_11.0 cusolver_dev_11.0 cusparse_11.0 cusparse_dev_11.0 npp_11.0 npp_dev_11.0 nvrtc_11.0 nvrtc_dev_11.0 nvml_dev_11.0"
)
if not exist "%SRC_DIR%\temp_build\cudnn-11.0-windows-x64-v8.0.2.39.zip" (
curl -k -L https://ossci-windows.s3.amazonaws.com/cudnn-11.0-windows-x64-v8.0.2.39.zip --output "%SRC_DIR%\temp_build\cudnn-11.0-windows-x64-v8.0.2.39.zip"
if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\cudnn-11.0-windows-x64-v8.0.2.39.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