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:
name: windows-gpu
environment:
<<: *environment
CUDA_VERSION: "11.3"
CUDA_VERSION: "11.6"
steps:
- checkout
- designate_upload_channel
......
......@@ -23,8 +23,8 @@ from jinja2 import select_autoescape
PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"]
CU_VERSIONS_DICT = {
"linux": ["cpu", "cu102", "cu113", "cu116", "cu117", "rocm5.0", "rocm5.1.1"],
"windows": ["cpu", "cu113", "cu116", "cu117"],
"linux": ["cpu", "cu102", "cu116", "cu117", "rocm5.0", "rocm5.1.1"],
"windows": ["cpu", "cu116", "cu117"],
"macos": ["cpu"],
}
......@@ -241,7 +241,7 @@ def unittest_workflows(indentation=6):
job = {
"name": f"unittest_{os_type}_{device_type}_py{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"],
}
......
......@@ -62,14 +62,6 @@ setup_cuda() {
fi
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)
if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2"
......@@ -261,9 +253,6 @@ setup_conda_cudatoolkit_constraint() {
cu116)
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.6 # [not osx]"
;;
cu113)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=11.3,<11.4 # [not osx]"
;;
cu102)
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