Unverified Commit 983d27ed authored by Nikita Shulga's avatar Nikita Shulga Committed by GitHub
Browse files

Move cmake_[windows|linux]_gpu to CUDA-11.3 (#4920)



* Set cuversion to 113 for cmake builds

* Build CUDA-11 Windows cmake using C++17

* Cherry-pick pkg_helper changes from pytorch/audio back to vision
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 81fe60cc
......@@ -1644,16 +1644,16 @@ workflows:
name: cmake_linux_cpu
python_version: '3.8'
- cmake_linux_gpu:
cu_version: cu102
cu_version: cu113
name: cmake_linux_gpu
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102
wheel_docker_image: pytorch/manylinux-cuda113
- cmake_windows_cpu:
cu_version: cpu
name: cmake_windows_cpu
python_version: '3.8'
- cmake_windows_gpu:
cu_version: cu102
cu_version: cu113
name: cmake_windows_gpu
python_version: '3.8'
- cmake_macos_cpu:
......
......@@ -268,9 +268,9 @@ def cmake_workflows(indentation=6):
for device in device_types:
job = {"name": f"cmake_{os_type}_{device}", "python_version": python_version}
job["cu_version"] = "cu102" if device == "gpu" else "cpu"
job["cu_version"] = "cu113" if device == "gpu" else "cpu"
if device == "gpu" and os_type == "linux":
job["wheel_docker_image"] = "pytorch/manylinux-cuda102"
job["wheel_docker_image"] = "pytorch/manylinux-cuda113"
jobs.append({f"cmake_{os_type}_{device}": job})
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