Skip to content

GitLab

Commit 2acafdaf authored by Andrey Talman's avatar Andrey Talman Committed by Facebook GitHub Bot
Browse files

CUDA 11.6 for TorchAudio (#2328) 

Summary:
CUDA 11.6 for TorchAudio

Pull Request resolved: https://github.com/pytorch/audio/pull/2328

Reviewed By: mthrok

Differential Revision: D35826414

Pulled By: atalman

fbshipit-source-id: 0a471f0566286d69c0c73191aea7fd5ac0647e5f
parent 6b242c29
Hide whitespace changes
Inline Side-by-side
Showing with 722 additions and 133 deletions
.circleci/config.yml
View file @ 2acafdaf
...@@ -756,6 +756,13 @@ workflows: ...@@ -756,6 +756,13 @@ workflows:
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cuda_version: cu116
name: binary_linux_wheel_py3.7_cu116
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.7_rocm4.3.1 name: binary_linux_wheel_py3.7_rocm4.3.1
...@@ -803,6 +810,13 @@ workflows: ...@@ -803,6 +810,13 @@ workflows:
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cuda_version: cu116
name: binary_linux_wheel_py3.8_cu116
python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.8_rocm4.3.1 name: binary_linux_wheel_py3.8_rocm4.3.1
...@@ -844,6 +858,13 @@ workflows: ...@@ -844,6 +858,13 @@ workflows:
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cuda_version: cu116
name: binary_linux_wheel_py3.9_cu116
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.9_rocm4.3.1 name: binary_linux_wheel_py3.9_rocm4.3.1
...@@ -885,6 +906,13 @@ workflows: ...@@ -885,6 +906,13 @@ workflows:
requires: requires:
- download_third_parties_nix - download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cuda_version: cu116
name: binary_linux_wheel_py3.10_cu116
python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda116
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.10_rocm4.3.1 name: binary_linux_wheel_py3.10_rocm4.3.1
...@@ -937,6 +965,11 @@ workflows: ...@@ -937,6 +965,11 @@ workflows:
name: binary_windows_wheel_py3.7_cu115 name: binary_windows_wheel_py3.7_cu115
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_windows_wheel:
cuda_version: cu116
name: binary_windows_wheel_py3.7_cu116
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda116
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
name: binary_windows_wheel_py3.8_cpu name: binary_windows_wheel_py3.8_cpu
...@@ -951,6 +984,11 @@ workflows: ...@@ -951,6 +984,11 @@ workflows:
name: binary_windows_wheel_py3.8_cu115 name: binary_windows_wheel_py3.8_cu115
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_windows_wheel:
cuda_version: cu116
name: binary_windows_wheel_py3.8_cu116
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda116
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
name: binary_windows_wheel_py3.9_cpu name: binary_windows_wheel_py3.9_cpu
...@@ -965,6 +1003,11 @@ workflows: ...@@ -965,6 +1003,11 @@ workflows:
name: binary_windows_wheel_py3.9_cu115 name: binary_windows_wheel_py3.9_cu115
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_windows_wheel:
cuda_version: cu116
name: binary_windows_wheel_py3.9_cu116
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda116
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
name: binary_windows_wheel_py3.10_cpu name: binary_windows_wheel_py3.10_cpu
...@@ -979,6 +1022,11 @@ workflows: ...@@ -979,6 +1022,11 @@ workflows:
name: binary_windows_wheel_py3.10_cu115 name: binary_windows_wheel_py3.10_cu115
python_version: '3.10' python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda115 wheel_docker_image: pytorch/manylinux-cuda115
- binary_windows_wheel:
cuda_version: cu116
name: binary_windows_wheel_py3.10_cu116
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda116
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1007,6 +1055,13 @@ workflows: ...@@ -1007,6 +1055,13 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
name: binary_linux_conda_py3.7_cu116
python_version: '3.7'
requires:
- download_third_parties_nix
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1035,6 +1090,13 @@ workflows: ...@@ -1035,6 +1090,13 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
name: binary_linux_conda_py3.8_cu116
python_version: '3.8'
requires:
- download_third_parties_nix
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1063,6 +1125,13 @@ workflows: ...@@ -1063,6 +1125,13 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
name: binary_linux_conda_py3.9_cu116
python_version: '3.9'
requires:
- download_third_parties_nix
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1091,6 +1160,13 @@ workflows: ...@@ -1091,6 +1160,13 @@ workflows:
python_version: '3.10' python_version: '3.10'
requires: requires:
- download_third_parties_nix - download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
name: binary_linux_conda_py3.10_cu116
python_version: '3.10'
requires:
- download_third_parties_nix
- binary_macos_conda: - binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1134,6 +1210,11 @@ workflows: ...@@ -1134,6 +1210,11 @@ workflows:
cuda_version: cu115 cuda_version: cu115
name: binary_windows_conda_py3.7_cu115 name: binary_windows_conda_py3.7_cu115
python_version: '3.7' python_version: '3.7'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
name: binary_windows_conda_py3.7_cu116
python_version: '3.7'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1149,6 +1230,11 @@ workflows: ...@@ -1149,6 +1230,11 @@ workflows:
cuda_version: cu115 cuda_version: cu115
name: binary_windows_conda_py3.8_cu115 name: binary_windows_conda_py3.8_cu115
python_version: '3.8' python_version: '3.8'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
name: binary_windows_conda_py3.8_cu116
python_version: '3.8'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1164,6 +1250,11 @@ workflows: ...@@ -1164,6 +1250,11 @@ workflows:
cuda_version: cu115 cuda_version: cu115
name: binary_windows_conda_py3.9_cu115 name: binary_windows_conda_py3.9_cu115
python_version: '3.9' python_version: '3.9'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
name: binary_windows_conda_py3.9_cu116
python_version: '3.9'
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -1179,6 +1270,11 @@ workflows: ...@@ -1179,6 +1270,11 @@ workflows:
cuda_version: cu115 cuda_version: cu115
name: binary_windows_conda_py3.10_cu115 name: binary_windows_conda_py3.10_cu115
python_version: '3.10' python_version: '3.10'
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
name: binary_windows_conda_py3.10_cu116
python_version: '3.10'
- build_docs: - build_docs:
filters: filters:
branches: branches:
...@@ -1438,6 +1534,43 @@ workflows: ...@@ -1438,6 +1534,43 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_wheel_py3.7_cu115_upload - nightly_binary_linux_wheel_py3.7_cu115_upload
- binary_linux_wheel:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu116
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda116
- 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_cu116_upload
requires:
- nightly_binary_linux_wheel_py3.7_cu116
subfolder: cu116/
- smoke_test_linux_pip:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu116_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_cu116_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm4.3.1
filters: filters:
...@@ -1659,6 +1792,43 @@ workflows: ...@@ -1659,6 +1792,43 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_wheel_py3.8_cu115_upload - nightly_binary_linux_wheel_py3.8_cu115_upload
- binary_linux_wheel:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu116
python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda116
- 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_cu116_upload
requires:
- nightly_binary_linux_wheel_py3.8_cu116
subfolder: cu116/
- smoke_test_linux_pip:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu116_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_cu116_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm4.3.1
filters: filters:
...@@ -1880,6 +2050,43 @@ workflows: ...@@ -1880,6 +2050,43 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_wheel_py3.9_cu115_upload - nightly_binary_linux_wheel_py3.9_cu115_upload
- binary_linux_wheel:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu116
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda116
- 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_cu116_upload
requires:
- nightly_binary_linux_wheel_py3.9_cu116
subfolder: cu116/
- smoke_test_linux_pip:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu116_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_cu116_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm4.3.1
filters: filters:
...@@ -2101,6 +2308,43 @@ workflows: ...@@ -2101,6 +2308,43 @@ workflows:
python_version: '3.10' python_version: '3.10'
requires: requires:
- nightly_binary_linux_wheel_py3.10_cu115_upload - nightly_binary_linux_wheel_py3.10_cu115_upload
- binary_linux_wheel:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_cu116
python_version: '3.10'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda116
- 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.10_cu116_upload
requires:
- nightly_binary_linux_wheel_py3.10_cu116
subfolder: cu116/
- smoke_test_linux_pip:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_cu116_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_linux_wheel_py3.10_cu116_upload
- binary_linux_wheel: - binary_linux_wheel:
cuda_version: rocm4.3.1 cuda_version: rocm4.3.1
filters: filters:
...@@ -2375,6 +2619,41 @@ workflows: ...@@ -2375,6 +2619,41 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_windows_wheel_py3.7_cu115_upload - nightly_binary_windows_wheel_py3.7_cu115_upload
- binary_windows_wheel:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu116
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda116
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu116_upload
requires:
- nightly_binary_windows_wheel_py3.7_cu116
subfolder: cu116/
- smoke_test_windows_pip:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.7_cu116_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_windows_wheel_py3.7_cu116_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2479,6 +2758,41 @@ workflows: ...@@ -2479,6 +2758,41 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_windows_wheel_py3.8_cu115_upload - nightly_binary_windows_wheel_py3.8_cu115_upload
- binary_windows_wheel:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu116
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda116
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu116_upload
requires:
- nightly_binary_windows_wheel_py3.8_cu116
subfolder: cu116/
- smoke_test_windows_pip:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.8_cu116_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_windows_wheel_py3.8_cu116_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2583,6 +2897,41 @@ workflows: ...@@ -2583,6 +2897,41 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_windows_wheel_py3.9_cu115_upload - nightly_binary_windows_wheel_py3.9_cu115_upload
- binary_windows_wheel:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu116
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda116
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu116_upload
requires:
- nightly_binary_windows_wheel_py3.9_cu116
subfolder: cu116/
- smoke_test_windows_pip:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.9_cu116_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_windows_wheel_py3.9_cu116_upload
- binary_windows_wheel: - binary_windows_wheel:
cuda_version: cpu cuda_version: cpu
filters: filters:
...@@ -2687,6 +3036,41 @@ workflows: ...@@ -2687,6 +3036,41 @@ workflows:
python_version: '3.10' python_version: '3.10'
requires: requires:
- nightly_binary_windows_wheel_py3.10_cu115_upload - nightly_binary_windows_wheel_py3.10_cu115_upload
- binary_windows_wheel:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.10_cu116
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda116
- binary_wheel_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.10_cu116_upload
requires:
- nightly_binary_windows_wheel_py3.10_cu116
subfolder: cu116/
- smoke_test_windows_pip:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_wheel_py3.10_cu116_smoke_test_pip
python_version: '3.10'
requires:
- nightly_binary_windows_wheel_py3.10_cu116_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -2831,6 +3215,42 @@ workflows: ...@@ -2831,6 +3215,42 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_linux_conda_py3.7_cu115_upload - nightly_binary_linux_conda_py3.7_cu115_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu116
python_version: '3.7'
requires:
- download_third_parties_nix
- 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_cu116_upload
requires:
- nightly_binary_linux_conda_py3.7_cu116
- smoke_test_linux_conda_gpu:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu116_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_linux_conda_py3.7_cu116_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -2975,6 +3395,42 @@ workflows: ...@@ -2975,6 +3395,42 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_linux_conda_py3.8_cu115_upload - nightly_binary_linux_conda_py3.8_cu115_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu116
python_version: '3.8'
requires:
- download_third_parties_nix
- 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_cu116_upload
requires:
- nightly_binary_linux_conda_py3.8_cu116
- smoke_test_linux_conda_gpu:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu116_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_linux_conda_py3.8_cu116_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -3119,6 +3575,42 @@ workflows: ...@@ -3119,6 +3575,42 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_linux_conda_py3.9_cu115_upload - nightly_binary_linux_conda_py3.9_cu115_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu116
python_version: '3.9'
requires:
- download_third_parties_nix
- 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_cu116_upload
requires:
- nightly_binary_linux_conda_py3.9_cu116
- smoke_test_linux_conda_gpu:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu116_smoke_test_conda
python_version: '3.9'
requires:
- nightly_binary_linux_conda_py3.9_cu116_upload
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -3263,6 +3755,42 @@ workflows: ...@@ -3263,6 +3755,42 @@ workflows:
python_version: '3.10' python_version: '3.10'
requires: requires:
- nightly_binary_linux_conda_py3.10_cu115_upload - nightly_binary_linux_conda_py3.10_cu115_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu116
python_version: '3.10'
requires:
- download_third_parties_nix
- 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.10_cu116_upload
requires:
- nightly_binary_linux_conda_py3.10_cu116
- smoke_test_linux_conda_gpu:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu116_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_linux_conda_py3.10_cu116_upload
- binary_macos_conda: - binary_macos_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -3461,6 +3989,40 @@ workflows: ...@@ -3461,6 +3989,40 @@ workflows:
python_version: '3.7' python_version: '3.7'
requires: requires:
- nightly_binary_windows_conda_py3.7_cu115_upload - nightly_binary_windows_conda_py3.7_cu115_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.7_cu116
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_windows_conda_py3.7_cu116_upload
requires:
- nightly_binary_windows_conda_py3.7_cu116
- smoke_test_windows_conda:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.7_cu116_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_windows_conda_py3.7_cu116_upload
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -3563,6 +4125,40 @@ workflows: ...@@ -3563,6 +4125,40 @@ workflows:
python_version: '3.8' python_version: '3.8'
requires: requires:
- nightly_binary_windows_conda_py3.8_cu115_upload - nightly_binary_windows_conda_py3.8_cu115_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.8_cu116
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_windows_conda_py3.8_cu116_upload
requires:
- nightly_binary_windows_conda_py3.8_cu116
- smoke_test_windows_conda:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.8_cu116_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_windows_conda_py3.8_cu116_upload
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -3665,6 +4261,40 @@ workflows: ...@@ -3665,6 +4261,40 @@ workflows:
python_version: '3.9' python_version: '3.9'
requires: requires:
- nightly_binary_windows_conda_py3.9_cu115_upload - nightly_binary_windows_conda_py3.9_cu115_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.9_cu116
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_windows_conda_py3.9_cu116_upload
requires:
- nightly_binary_windows_conda_py3.9_cu116
- smoke_test_windows_conda:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.9_cu116_smoke_test_conda
python_version: '3.9'
requires:
- nightly_binary_windows_conda_py3.9_cu116_upload
- binary_windows_conda: - binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cpu conda_docker_image: pytorch/conda-builder:cpu
cuda_version: cpu cuda_version: cpu
...@@ -3767,3 +4397,37 @@ workflows: ...@@ -3767,3 +4397,37 @@ workflows:
python_version: '3.10' python_version: '3.10'
requires: requires:
- nightly_binary_windows_conda_py3.10_cu115_upload - nightly_binary_windows_conda_py3.10_cu115_upload
- binary_windows_conda:
conda_docker_image: pytorch/conda-builder:cuda116
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cu116
python_version: '3.10'
- binary_conda_upload:
context: org-member
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cu116_upload
requires:
- nightly_binary_windows_conda_py3.10_cu116
- smoke_test_windows_conda:
cuda_version: cu116
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_windows_conda_py3.10_cu116_smoke_test_conda
python_version: '3.10'
requires:
- nightly_binary_windows_conda_py3.10_cu116_upload
.circleci/regenerate.py
View file @ 2acafdaf
...@@ -23,8 +23,8 @@ from jinja2 import select_autoescape ...@@ -23,8 +23,8 @@ from jinja2 import select_autoescape
PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"] PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
CU_VERSIONS_DICT = { CU_VERSIONS_DICT = {
"linux": ["cpu", "cu102", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"], "linux": ["cpu", "cu102", "cu113", "cu115", "cu116", "rocm4.3.1", "rocm4.5.2"],
"windows": ["cpu", "cu113", "cu115"], "windows": ["cpu", "cu113", "cu115", "cu116"],
"macos": ["cpu"], "macos": ["cpu"],
} }
......
packaging/build_conda.sh
View file @ 2acafdaf
...@@ -16,7 +16,7 @@ setup_visual_studio_constraint ...@@ -16,7 +16,7 @@ setup_visual_studio_constraint
export CUDATOOLKIT_CHANNEL="nvidia" export CUDATOOLKIT_CHANNEL="nvidia"
# NOTE: This is needed because `cudatoolkit=11.5` has a dependency on conda-forge # NOTE: This is needed because `cudatoolkit=11.5` has a dependency on conda-forge
# See: https://github.com/pytorch/audio/pull/2224#issuecomment-1049185550 # See: https://github.com/pytorch/audio/pull/2224#issuecomment-1049185550
if [[ ${CU_VERSION} = "cu115" ]]; then if [[ ${CU_VERSION} = "cu115" || ${CU_VERSION} = "cu116" ]]; then
CONDA_CHANNEL_FLAGS="${CONDA_CHANNEL_FLAGS} -c conda-forge" CONDA_CHANNEL_FLAGS="${CONDA_CHANNEL_FLAGS} -c conda-forge"
fi fi
# NOTE: There are some dependencies that are not available for macOS on Python 3.10 without conda-forge # NOTE: There are some dependencies that are not available for macOS on Python 3.10 without conda-forge
......
packaging/pkg_helpers.bash
View file @ 2acafdaf
...@@ -46,6 +46,14 @@ setup_cuda() { ...@@ -46,6 +46,14 @@ setup_cuda() {
# Now work out the CUDA settings # Now work out the CUDA settings
case "$CU_VERSION" in case "$CU_VERSION" in
cu116)
if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.6"
else
export CUDA_HOME=/usr/local/cuda-11.6/
fi
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
;;
cu115) cu115)
if [[ "$OSTYPE" == "msys" ]]; then if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.5" export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.5"
...@@ -261,33 +269,18 @@ setup_conda_cudatoolkit_constraint() { ...@@ -261,33 +269,18 @@ setup_conda_cudatoolkit_constraint() {
export CONDA_BUILD_VARIANT="cpu" export CONDA_BUILD_VARIANT="cpu"
else else
case "$CU_VERSION" in case "$CU_VERSION" in
cu116)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.6,<11.7 # [not osx]"
;;
cu115) cu115)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.5,<11.6 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.5,<11.6 # [not osx]"
;; ;;
cu113) cu113)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.3,<11.4 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.3,<11.4 # [not osx]"
;; ;;
cu112)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.2,<11.3 # [not osx]"
;;
cu111)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.1,<11.2 # [not osx]"
;;
cu110)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.0,<11.1 # [not osx]"
;;
cu102) cu102)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=10.2,<10.3 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=10.2,<10.3 # [not osx]"
;; ;;
cu101)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=10.1,<10.2 # [not osx]"
;;
cu100)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=10.0,<10.1 # [not osx]"
;;
cu92)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=9.2,<9.3 # [not osx]"
;;
cpu) cpu)
export CONDA_CUDATOOLKIT_CONSTRAINT="" export CONDA_CUDATOOLKIT_CONSTRAINT=""
export CONDA_BUILD_VARIANT="cpu" export CONDA_BUILD_VARIANT="cpu"
......
packaging/windows/internal/cuda_install.bat
View file @ 2acafdaf
...@@ -20,70 +20,17 @@ set /a CUDA_VER=%CU_VERSION:cu=% ...@@ -20,70 +20,17 @@ set /a CUDA_VER=%CU_VERSION:cu=%
set CUDA_VER_MAJOR=%CUDA_VER:~0,-1% set CUDA_VER_MAJOR=%CUDA_VER:~0,-1%
set CUDA_VER_MINOR=%CUDA_VER:~-1,1% set CUDA_VER_MINOR=%CUDA_VER:~-1,1%
set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR% set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR%
set CUDNN_FOLDER="cuda"
set CUDNN_LIB_FOLDER="lib\x64"
if %CUDA_VER% EQU 92 goto cuda92
if %CUDA_VER% EQU 100 goto cuda100
if %CUDA_VER% EQU 101 goto cuda101
if %CUDA_VER% EQU 102 goto cuda102 if %CUDA_VER% EQU 102 goto cuda102
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 113 goto cuda113 if %CUDA_VER% EQU 113 goto cuda113
if %CUDA_VER% EQU 115 goto cuda115 if %CUDA_VER% EQU 115 goto cuda115
if %CUDA_VER% EQU 116 goto cuda116
echo CUDA %CUDA_VERSION_STR% is not supported echo CUDA %CUDA_VERSION_STR% is not supported
exit /b 1 exit /b 1
:cuda92
if not exist "%SRC_DIR%\temp_build\cuda_9.2.148_win10.exe" (
curl -k -L https://ossci-windows.s3.amazonaws.com/win2016/cuda_9.2.148_win10.exe --output "%SRC_DIR%\temp_build\cuda_9.2.148_win10.exe"
if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\cuda_9.2.148_win10.exe"
set "ARGS=nvcc_9.2 cuobjdump_9.2 nvprune_9.2 cupti_9.2 cublas_9.2 cublas_dev_9.2 cudart_9.2 cufft_9.2 cufft_dev_9.2 curand_9.2 curand_dev_9.2 cusolver_9.2 cusolver_dev_9.2 cusparse_9.2 cusparse_dev_9.2 nvgraph_9.2 nvgraph_dev_9.2 npp_9.2 npp_dev_9.2 nvrtc_9.2 nvrtc_dev_9.2 nvml_dev_9.2"
)
if not exist "%SRC_DIR%\temp_build\cudnn-9.2-windows10-x64-v7.2.1.38.zip" (
curl -k -L https://ossci-windows.s3.amazonaws.com/win2016/cudnn-9.2-windows10-x64-v7.2.1.38.zip --output "%SRC_DIR%\temp_build\cudnn-9.2-windows10-x64-v7.2.1.38.zip"
if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\cudnn-9.2-windows10-x64-v7.2.1.38.zip"
)
goto cuda_common
:cuda100
if not exist "%SRC_DIR%\temp_build\cuda_10.0.130_411.31_win10.exe" (
curl -k -L https://ossci-windows.s3.amazonaws.com/win2016/cuda_10.0.130_411.31_win10.exe --output "%SRC_DIR%\temp_build\cuda_10.0.130_411.31_win10.exe"
if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\cuda_10.0.130_411.31_win10.exe"
set "ARGS=nvcc_10.0 cuobjdump_10.0 nvprune_10.0 cupti_10.0 cublas_10.0 cublas_dev_10.0 cudart_10.0 cufft_10.0 cufft_dev_10.0 curand_10.0 curand_dev_10.0 cusolver_10.0 cusolver_dev_10.0 cusparse_10.0 cusparse_dev_10.0 nvgraph_10.0 nvgraph_dev_10.0 npp_10.0 npp_dev_10.0 nvrtc_10.0 nvrtc_dev_10.0 nvml_dev_10.0"
)
if not exist "%SRC_DIR%\temp_build\cudnn-10.0-windows10-x64-v7.4.1.5.zip" (
curl -k -L https://ossci-windows.s3.amazonaws.com/win2016/cudnn-10.0-windows10-x64-v7.4.1.5.zip --output "%SRC_DIR%\temp_build\cudnn-10.0-windows10-x64-v7.4.1.5.zip"
if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\cudnn-10.0-windows10-x64-v7.4.1.5.zip"
)
goto cuda_common
:cuda101
if not exist "%SRC_DIR%\temp_build\cuda_10.1.243_426.00_win10.exe" (
curl -k -L https://ossci-windows.s3.amazonaws.com/cuda_10.1.243_426.00_win10.exe --output "%SRC_DIR%\temp_build\cuda_10.1.243_426.00_win10.exe"
if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\cuda_10.1.243_426.00_win10.exe"
set "ARGS=nvcc_10.1 cuobjdump_10.1 nvprune_10.1 cupti_10.1 cublas_10.1 cublas_dev_10.1 cudart_10.1 cufft_10.1 cufft_dev_10.1 curand_10.1 curand_dev_10.1 cusolver_10.1 cusolver_dev_10.1 cusparse_10.1 cusparse_dev_10.1 nvgraph_10.1 nvgraph_dev_10.1 npp_10.1 npp_dev_10.1 nvrtc_10.1 nvrtc_dev_10.1 nvml_dev_10.1"
)
if not exist "%SRC_DIR%\temp_build\cudnn-10.1-windows10-x64-v7.6.4.38.zip" (
curl -k -L https://ossci-windows.s3.amazonaws.com/cudnn-10.1-windows10-x64-v7.6.4.38.zip --output "%SRC_DIR%\temp_build\cudnn-10.1-windows10-x64-v7.6.4.38.zip"
if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\cudnn-10.1-windows10-x64-v7.6.4.38.zip"
)
goto cuda_common
:cuda102 :cuda102
if not exist "%SRC_DIR%\temp_build\cuda_10.2.89_441.22_win10.exe" ( if not exist "%SRC_DIR%\temp_build\cuda_10.2.89_441.22_win10.exe" (
...@@ -109,96 +56,81 @@ echo Installing GPU driver DLLs ...@@ -109,96 +56,81 @@ echo Installing GPU driver DLLs
goto cuda_common goto cuda_common
:cuda110 :cuda113
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.4.30.zip" (
curl -k -L https://ossci-windows.s3.amazonaws.com/cudnn-11.0-windows-x64-v8.0.4.30.zip --output "%SRC_DIR%\temp_build\cudnn-11.0-windows-x64-v8.0.4.30.zip"
if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\cudnn-11.0-windows-x64-v8.0.4.30.zip"
)
goto cuda_common
:cuda111
if not exist "%SRC_DIR%\temp_build\cuda_11.1.1_456.81_win10.exe" ( set CUDA_INSTALL_EXE=cuda_11.3.0_465.89_win10.exe
curl -k -L https://ossci-windows.s3.amazonaws.com/cuda_11.1.1_456.81_win10.exe --output "%SRC_DIR%\temp_build\cuda_11.1.1_456.81_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 if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\cuda_11.1.1_456.81_win10.exe" set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_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" 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"
)
@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 set CUDNN_INSTALL_ZIP=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive.zip
set CUDNN_FOLDER=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive
:cuda112 set CUDNN_LIB_FOLDER="lib"
if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" (
if not exist "%SRC_DIR%\temp_build\cuda_11.2.0_460.89_win10.exe" ( curl -k -L "http://s3.amazonaws.com/ossci-windows/%CUDNN_INSTALL_ZIP%" --output "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
curl -k -L https://ossci-windows.s3.amazonaws.com/cuda_11.2.0_460.89_win10.exe --output "%SRC_DIR%\temp_build\cuda_11.2.0_460.89_win10.exe"
if errorlevel 1 exit /b 1 if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\cuda_11.2.0_460.89_win10.exe" set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
set "ARGS=nvcc_11.2 cuobjdump_11.2 nvprune_11.2 nvprof_11.2 cupti_11.2 cublas_11.2 cublas_dev_11.2 cudart_11.2 cufft_11.2 cufft_dev_11.2 curand_11.2 curand_dev_11.2 cusolver_11.2 cusolver_dev_11.2 cusparse_11.2 cusparse_dev_11.2 npp_11.2 npp_dev_11.2 nvrtc_11.2 nvrtc_dev_11.2 nvml_dev_11.2"
)
if not exist "%SRC_DIR%\temp_build\cudnn-11.2-windows-x64-v8.1.0.77.zip" ( rem Make sure windows path contains zlib dll
curl -k -L http://s3.amazonaws.com/ossci-windows/cudnn-11.2-windows-x64-v8.1.0.77.zip --output "%SRC_DIR%\temp_build\cudnn-11.2-windows-x64-v8.1.0.77.zip" curl -k -L "http://s3.amazonaws.com/ossci-windows/zlib123dllx64.zip" --output "%SRC_DIR%\temp_build\zlib123dllx64.zip"
if errorlevel 1 exit /b 1 7z x "%SRC_DIR%\temp_build\zlib123dllx64.zip" -o"%SRC_DIR%\temp_build\zlib"
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\cudnn-11.2-windows-x64-v8.1.0.77.zip" xcopy /Y "%SRC_DIR%\temp_build\zlib\dll_x64\*.dll" "C:\Windows\System32"
) )
goto cuda_common goto cuda_common
:cuda113 :cuda115
set CUDA_INSTALL_EXE=cuda_11.3.0_465.89_win10.exe set CUDA_INSTALL_EXE=cuda_11.5.0_496.13_win10.exe
if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_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%" 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 if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" 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 "ARGS=thrust_11.5 nvcc_11.5 cuobjdump_11.5 nvprune_11.5 nvprof_11.5 cupti_11.5 cublas_11.5 cublas_dev_11.5 cudart_11.5 cufft_11.5 cufft_dev_11.5 curand_11.5 curand_dev_11.5 cusolver_11.5 cusolver_dev_11.5 cusparse_11.5 cusparse_dev_11.5 npp_11.5 npp_dev_11.5 nvrtc_11.5 nvrtc_dev_11.5 nvml_dev_11.5"
) )
set CUDNN_INSTALL_ZIP=cudnn-11.3-windows-x64-v8.2.0.53.zip set CUDNN_INSTALL_ZIP=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive.zip
set CUDNN_FOLDER=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive
set CUDNN_LIB_FOLDER="lib"
if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_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%" 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 if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
rem Make sure windows path contains zlib dll
curl -k -L "http://s3.amazonaws.com/ossci-windows/zlib123dllx64.zip" --output "%SRC_DIR%\temp_build\zlib123dllx64.zip"
7z x "%SRC_DIR%\temp_build\zlib123dllx64.zip" -o"%SRC_DIR%\temp_build\zlib"
xcopy /Y "%SRC_DIR%\temp_build\zlib\dll_x64\*.dll" "C:\Windows\System32"
) )
goto cuda_common goto cuda_common
:cuda115 :cuda116
set CUDA_INSTALL_EXE=cuda_11.5.0_496.13_win10.exe set CUDA_INSTALL_EXE=cuda_11.6.0_511.23_windows.exe
if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_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%" 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 if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
set "ARGS=thrust_11.5 nvcc_11.5 cuobjdump_11.5 nvprune_11.5 nvprof_11.5 cupti_11.5 cublas_11.5 cublas_dev_11.5 cudart_11.5 cufft_11.5 cufft_dev_11.5 curand_11.5 curand_dev_11.5 cusolver_11.5 cusolver_dev_11.5 cusparse_11.5 cusparse_dev_11.5 npp_11.5 npp_dev_11.5 nvrtc_11.5 nvrtc_dev_11.5 nvml_dev_11.5" set "ARGS=thrust_11.6 nvcc_11.6 cuobjdump_11.6 nvprune_11.6 nvprof_11.6 cupti_11.6 cublas_11.6 cublas_dev_11.6 cudart_11.6 cufft_11.6 cufft_dev_11.6 curand_11.6 curand_dev_11.6 cusolver_11.6 cusolver_dev_11.6 cusparse_11.6 cusparse_dev_11.6 npp_11.6 npp_dev_11.6 nvrtc_11.6 nvrtc_dev_11.6 nvml_dev_11.6"
) )
set CUDNN_INSTALL_ZIP=cudnn-11.3-windows-x64-v8.2.0.53.zip set CUDNN_INSTALL_ZIP=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive.zip
set CUDNN_FOLDER=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive
set CUDNN_LIB_FOLDER="lib"
if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_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%" 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 if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
rem Make sure windows path contains zlib dll
curl -k -L "http://s3.amazonaws.com/ossci-windows/zlib123dllx64.zip" --output "%SRC_DIR%\temp_build\zlib123dllx64.zip"
7z x "%SRC_DIR%\temp_build\zlib123dllx64.zip" -o"%SRC_DIR%\temp_build\zlib"
xcopy /Y "%SRC_DIR%\temp_build\zlib\dll_x64\*.dll" "C:\Windows\System32"
) )
goto cuda_common goto cuda_common
...@@ -256,9 +188,9 @@ if not exist "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_ST ...@@ -256,9 +188,9 @@ if not exist "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_ST
echo Installing cuDNN... echo Installing cuDNN...
7z x %CUDNN_SETUP_FILE% -o"%SRC_DIR%\temp_build\cudnn" 7z x %CUDNN_SETUP_FILE% -o"%SRC_DIR%\temp_build\cudnn"
xcopy /Y "%SRC_DIR%\temp_build\cudnn\cuda\bin\*.*" "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\bin" xcopy /Y "%SRC_DIR%\temp_build\cudnn\%CUDNN_FOLDER%\bin\*.*" "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\bin"
xcopy /Y "%SRC_DIR%\temp_build\cudnn\cuda\lib\x64\*.*" "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\lib\x64" xcopy /Y "%SRC_DIR%\temp_build\cudnn\%CUDNN_FOLDER%\%CUDNN_LIB_FOLDER%\*.*" "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\lib\x64"
xcopy /Y "%SRC_DIR%\temp_build\cudnn\cuda\include\*.*" "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\include" xcopy /Y "%SRC_DIR%\temp_build\cudnn\%CUDNN_FOLDER%\include\*.*" "%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v%CUDA_VERSION_STR%\include"
echo Cleaning temp files echo Cleaning temp files
rd /s /q "%SRC_DIR%\temp_build" || ver > nul rd /s /q "%SRC_DIR%\temp_build" || ver > nul
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