Unverified Commit ccc183da authored by Yi Zhang's avatar Yi Zhang Committed by GitHub
Browse files

Use cu113 for unittest_windows_gpu (#1853)



* set cu113 for unittest_windows_gpu

* fix old logic

* Update .circleci/regenerate.py
Co-authored-by: default avatarNikita Shulga <nikita.shulga@gmail.com>
parent a85f1adc
...@@ -517,7 +517,7 @@ jobs: ...@@ -517,7 +517,7 @@ jobs:
name: windows-gpu name: windows-gpu
environment: environment:
<<: *environment <<: *environment
CUDA_VERSION: "10.2" CUDA_VERSION: "11.3"
TORCHAUDIO_TEST_FORCE_CUDA: 1 TORCHAUDIO_TEST_FORCE_CUDA: 1
steps: steps:
- checkout - checkout
......
...@@ -517,7 +517,7 @@ jobs: ...@@ -517,7 +517,7 @@ jobs:
name: windows-gpu name: windows-gpu
environment: environment:
<<: *environment <<: *environment
CUDA_VERSION: "10.2" CUDA_VERSION: "11.3"
TORCHAUDIO_TEST_FORCE_CUDA: 1 TORCHAUDIO_TEST_FORCE_CUDA: 1
steps: steps:
- checkout - checkout
......
...@@ -222,7 +222,7 @@ def unittest_workflows(indentation=6): ...@@ -222,7 +222,7 @@ def unittest_workflows(indentation=6):
"stylecheck": { "stylecheck": {
"name": f"stylecheck_py{python_version}", "name": f"stylecheck_py{python_version}",
"python_version": python_version, "python_version": python_version,
"cuda_version": 'cpu' if device_type=="cpu" else "cu113", "cuda_version": "cpu",
} }
}) })
return indent(indentation, jobs) return indent(indentation, jobs)
......
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