Commit 155dc298 authored by Andrey Talman's avatar Andrey Talman Committed by Facebook GitHub Bot
Browse files

CUDA 11.3 remove. New Stable version is 11.6 (#2670)

Summary:
CUDA 11.3 Removing.

Core PR: https://github.com/pytorch/pytorch/pull/84866
cc malfet ptrblck

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

Reviewed By: malfet, osalpekar

Differential Revision: D39449263

Pulled By: atalman

fbshipit-source-id: f86bb119685ead3ffcabd92c4bb8076aecde4095
parent 4d535e88
This diff is collapsed.
...@@ -715,7 +715,7 @@ jobs: ...@@ -715,7 +715,7 @@ jobs:
name: windows-gpu name: windows-gpu
environment: environment:
<<: *environment <<: *environment
CUDA_VERSION: "11.3" CUDA_VERSION: "11.6"
steps: steps:
- checkout - checkout
- designate_upload_channel - designate_upload_channel
......
...@@ -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", "cu116", "cu117", "rocm5.0", "rocm5.1.1"], "linux": ["cpu", "cu102", "cu116", "cu117", "rocm5.0", "rocm5.1.1"],
"windows": ["cpu", "cu113", "cu116", "cu117"], "windows": ["cpu", "cu116", "cu117"],
"macos": ["cpu"], "macos": ["cpu"],
} }
...@@ -241,7 +241,7 @@ def unittest_workflows(indentation=6): ...@@ -241,7 +241,7 @@ def unittest_workflows(indentation=6):
job = { job = {
"name": f"unittest_{os_type}_{device_type}_py{python_version}", "name": f"unittest_{os_type}_{device_type}_py{python_version}",
"python_version": python_version, "python_version": python_version,
"cuda_version": "cpu" if device_type == "cpu" else "cu113", "cuda_version": "cpu" if device_type == "cpu" else "cu116",
"requires": ["download_third_parties"], "requires": ["download_third_parties"],
} }
......
...@@ -62,14 +62,6 @@ setup_cuda() { ...@@ -62,14 +62,6 @@ setup_cuda() {
fi fi
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
;; ;;
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 TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
;;
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"
...@@ -261,9 +253,6 @@ setup_conda_cudatoolkit_constraint() { ...@@ -261,9 +253,6 @@ setup_conda_cudatoolkit_constraint() {
cu116) cu116)
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.6 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.6 # [not osx]"
;; ;;
cu113)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.3,<11.4 # [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]"
;; ;;
......
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