Skip to content

GitLab

".github/vscode:/vscode.git/clone" did not exist on "3cef23250cff1515bdd169a1ee03c2203fe580e7"
Commit bb73934f authored by Nikita Shulga's avatar Nikita Shulga Committed by Facebook GitHub Bot
Browse files

Update ROCM version 4.1 -> 4.3.1 and 4.5 (#2186) 

Summary:
Also, retire cuda-10.2

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

Reviewed By: mthrok

Differential Revision: D33917595

Pulled By: malfet

fbshipit-source-id: 060d3fa706279fe45ffd1f4f99e5727520612d56
parent 157cb2a2
Show whitespace changes
Inline Side-by-side
Showing with 132 additions and 261 deletions
.circleci/config.yml
View file @ bb73934f
......@@ -725,13 +725,6 @@ workflows:
python_version: '3.7'
requires:
- download_third_parties_nix
- binary_linux_wheel:
cuda_version: cu102
name: binary_linux_wheel_py3.7_cu102
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
cuda_version: cu111
name: binary_linux_wheel_py3.7_cu111
......@@ -754,12 +747,19 @@ workflows:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cuda_version: rocm4.1
name: binary_linux_wheel_py3.7_rocm4.1
cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.7_rocm4.3.1
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel:
cuda_version: rocm4.5.2
name: binary_linux_wheel_py3.7_rocm4.5.2
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_linux_wheel:
cuda_version: cpu
filters:
......@@ -772,13 +772,6 @@ workflows:
python_version: '3.8'
requires:
- download_third_parties_nix
- binary_linux_wheel:
cuda_version: cu102
name: binary_linux_wheel_py3.8_cu102
python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
cuda_version: cu111
name: binary_linux_wheel_py3.8_cu111
......@@ -801,25 +794,25 @@ workflows:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cuda_version: rocm4.1
name: binary_linux_wheel_py3.8_rocm4.1
cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.8_rocm4.3.1
python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel:
cuda_version: cpu
name: binary_linux_wheel_py3.9_cpu
python_version: '3.9'
cuda_version: rocm4.5.2
name: binary_linux_wheel_py3.8_rocm4.5.2
python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_linux_wheel:
cuda_version: cu102
name: binary_linux_wheel_py3.9_cu102
cuda_version: cpu
name: binary_linux_wheel_py3.9_cpu
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel:
cuda_version: cu111
name: binary_linux_wheel_py3.9_cu111
......@@ -842,12 +835,19 @@ workflows:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda115
- binary_linux_wheel:
cuda_version: rocm4.1
name: binary_linux_wheel_py3.9_rocm4.1
cuda_version: rocm4.3.1
name: binary_linux_wheel_py3.9_rocm4.3.1
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_linux_wheel:
cuda_version: rocm4.5.2
name: binary_linux_wheel_py3.9_rocm4.5.2
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_macos_wheel:
cuda_version: cpu
name: binary_macos_wheel_py3.7_cpu
......@@ -915,13 +915,6 @@ workflows:
python_version: '3.7'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
name: binary_linux_conda_py3.7_cu102
python_version: '3.7'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
......@@ -950,13 +943,6 @@ workflows:
python_version: '3.8'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
name: binary_linux_conda_py3.8_cu102
python_version: '3.8'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
......@@ -985,13 +971,6 @@ workflows:
python_version: '3.9'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
name: binary_linux_conda_py3.9_cu102
python_version: '3.9'
requires:
- download_third_parties_nix
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
......@@ -1243,43 +1222,6 @@ workflows:
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_cpu_upload
- binary_linux_wheel:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu102
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- 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_cu102_upload
requires:
- nightly_binary_linux_wheel_py3.7_cu102
subfolder: cu102/
- smoke_test_linux_pip:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu102_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_cu102_upload
- binary_linux_wheel:
cuda_version: cu111
filters:
......@@ -1392,18 +1334,18 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.7_cu115_upload
- binary_linux_wheel:
cuda_version: rocm4.1
cuda_version: rocm4.3.1
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.1
name: nightly_binary_linux_wheel_py3.7_rocm4.3.1
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -1412,34 +1354,35 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.1_upload
name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.1
subfolder: rocm4.1/
- nightly_binary_linux_wheel_py3.7_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
cuda_version: rocm4.1
cuda_version: rocm4.3.1
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_rocm4.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.7_rocm4.3.1_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.7_rocm4.1_upload
- nightly_binary_linux_wheel_py3.7_rocm4.3.1_upload
- binary_linux_wheel:
cuda_version: cpu
cuda_version: rocm4.5.2
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cpu
python_version: '3.8'
name: nightly_binary_linux_wheel_py3.7_rocm4.5.2
python_version: '3.7'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -1448,35 +1391,34 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cpu_upload
name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
requires:
- nightly_binary_linux_wheel_py3.8_cpu
subfolder: cpu/
- nightly_binary_linux_wheel_py3.7_rocm4.5.2
subfolder: rocm4.5.2/
- smoke_test_linux_pip:
cuda_version: cpu
cuda_version: rocm4.5.2
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cpu_smoke_test_pip
python_version: '3.8'
name: nightly_binary_linux_wheel_py3.7_rocm4.5.2_smoke_test_pip
python_version: '3.7'
requires:
- nightly_binary_linux_wheel_py3.8_cpu_upload
- nightly_binary_linux_wheel_py3.7_rocm4.5.2_upload
- binary_linux_wheel:
cuda_version: cu102
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu102
name: nightly_binary_linux_wheel_py3.8_cpu
python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload:
context: org-member
filters:
......@@ -1485,22 +1427,22 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu102_upload
name: nightly_binary_linux_wheel_py3.8_cpu_upload
requires:
- nightly_binary_linux_wheel_py3.8_cu102
subfolder: cu102/
- nightly_binary_linux_wheel_py3.8_cpu
subfolder: cpu/
- smoke_test_linux_pip:
cuda_version: cu102
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu102_smoke_test_pip
name: nightly_binary_linux_wheel_py3.8_cpu_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_cu102_upload
- nightly_binary_linux_wheel_py3.8_cpu_upload
- binary_linux_wheel:
cuda_version: cu111
filters:
......@@ -1613,18 +1555,18 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.8_cu115_upload
- binary_linux_wheel:
cuda_version: rocm4.1
cuda_version: rocm4.3.1
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.1
name: nightly_binary_linux_wheel_py3.8_rocm4.3.1
python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- binary_wheel_upload:
context: org-member
filters:
......@@ -1633,34 +1575,35 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.1_upload
name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.1
subfolder: rocm4.1/
- nightly_binary_linux_wheel_py3.8_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
cuda_version: rocm4.1
cuda_version: rocm4.3.1
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_rocm4.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.8_rocm4.3.1_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.8_rocm4.1_upload
- nightly_binary_linux_wheel_py3.8_rocm4.3.1_upload
- binary_linux_wheel:
cuda_version: cpu
cuda_version: rocm4.5.2
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cpu
python_version: '3.9'
name: nightly_binary_linux_wheel_py3.8_rocm4.5.2
python_version: '3.8'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -1669,35 +1612,34 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cpu_upload
name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
requires:
- nightly_binary_linux_wheel_py3.9_cpu
subfolder: cpu/
- nightly_binary_linux_wheel_py3.8_rocm4.5.2
subfolder: rocm4.5.2/
- smoke_test_linux_pip:
cuda_version: cpu
cuda_version: rocm4.5.2
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cpu_smoke_test_pip
python_version: '3.9'
name: nightly_binary_linux_wheel_py3.8_rocm4.5.2_smoke_test_pip
python_version: '3.8'
requires:
- nightly_binary_linux_wheel_py3.9_cpu_upload
- nightly_binary_linux_wheel_py3.8_rocm4.5.2_upload
- binary_linux_wheel:
cuda_version: cu102
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu102
name: nightly_binary_linux_wheel_py3.9_cpu
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload:
context: org-member
filters:
......@@ -1706,22 +1648,22 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu102_upload
name: nightly_binary_linux_wheel_py3.9_cpu_upload
requires:
- nightly_binary_linux_wheel_py3.9_cu102
subfolder: cu102/
- nightly_binary_linux_wheel_py3.9_cpu
subfolder: cpu/
- smoke_test_linux_pip:
cuda_version: cu102
cuda_version: cpu
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu102_smoke_test_pip
name: nightly_binary_linux_wheel_py3.9_cpu_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_cu102_upload
- nightly_binary_linux_wheel_py3.9_cpu_upload
- binary_linux_wheel:
cuda_version: cu111
filters:
......@@ -1834,18 +1776,55 @@ workflows:
requires:
- nightly_binary_linux_wheel_py3.9_cu115_upload
- binary_linux_wheel:
cuda_version: rocm4.1
cuda_version: rocm4.3.1
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.3.1
- 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_rocm4.3.1_upload
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.3.1
subfolder: rocm4.3.1/
- smoke_test_linux_pip:
cuda_version: rocm4.3.1
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.3.1_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.3.1_upload
- binary_linux_wheel:
cuda_version: rocm4.5.2
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.1
name: nightly_binary_linux_wheel_py3.9_rocm4.5.2
python_version: '3.9'
requires:
- download_third_parties_nix
wheel_docker_image: pytorch/manylinux-rocm:4.1
wheel_docker_image: pytorch/manylinux-rocm:4.5.2
- binary_wheel_upload:
context: org-member
filters:
......@@ -1854,22 +1833,22 @@ workflows:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.1_upload
name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.1
subfolder: rocm4.1/
- nightly_binary_linux_wheel_py3.9_rocm4.5.2
subfolder: rocm4.5.2/
- smoke_test_linux_pip:
cuda_version: rocm4.1
cuda_version: rocm4.5.2
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_rocm4.1_smoke_test_pip
name: nightly_binary_linux_wheel_py3.9_rocm4.5.2_smoke_test_pip
python_version: '3.9'
requires:
- nightly_binary_linux_wheel_py3.9_rocm4.1_upload
- nightly_binary_linux_wheel_py3.9_rocm4.5.2_upload
- binary_macos_wheel:
cuda_version: cpu
filters:
......@@ -2290,42 +2269,6 @@ workflows:
python_version: '3.7'
requires:
- nightly_binary_linux_conda_py3.7_cpu_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu102
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_cu102_upload
requires:
- nightly_binary_linux_conda_py3.7_cu102
- smoke_test_linux_conda_gpu:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu102_smoke_test_conda
python_version: '3.7'
requires:
- nightly_binary_linux_conda_py3.7_cu102_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
......@@ -2470,42 +2413,6 @@ workflows:
python_version: '3.8'
requires:
- nightly_binary_linux_conda_py3.8_cpu_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu102
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_cu102_upload
requires:
- nightly_binary_linux_conda_py3.8_cu102
- smoke_test_linux_conda_gpu:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu102_smoke_test_conda
python_version: '3.8'
requires:
- nightly_binary_linux_conda_py3.8_cu102_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
......@@ -2650,42 +2557,6 @@ workflows:
python_version: '3.9'
requires:
- nightly_binary_linux_conda_py3.9_cpu_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu102
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_cu102_upload
requires:
- nightly_binary_linux_conda_py3.9_cu102
- smoke_test_linux_conda_gpu:
cuda_version: cu102
filters:
branches:
only:
- nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu102_smoke_test_conda
python_version: '3.9'
requires:
- nightly_binary_linux_conda_py3.9_cu102_upload
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda111
cuda_version: cu111
......
.circleci/regenerate.py
View file @ bb73934f
......@@ -23,7 +23,7 @@ from jinja2 import select_autoescape
PYTHON_VERSIONS = ["3.7", "3.8", "3.9"]
CU_VERSIONS_DICT = {
"linux": ["cpu", "cu102", "cu111", "cu113", "cu115", "rocm4.1"],
"linux": ["cpu", "cu111", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
"windows": ["cpu", "cu113", "cu115"],
"macos": ["cpu"],
}
......
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