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: ...@@ -1644,16 +1644,16 @@ workflows:
name: cmake_linux_cpu name: cmake_linux_cpu
python_version: '3.8' python_version: '3.8'
- cmake_linux_gpu: - cmake_linux_gpu:
cu_version: cu102 cu_version: cu113
name: cmake_linux_gpu name: cmake_linux_gpu
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102 wheel_docker_image: pytorch/manylinux-cuda113
- cmake_windows_cpu: - cmake_windows_cpu:
cu_version: cpu cu_version: cpu
name: cmake_windows_cpu name: cmake_windows_cpu
python_version: '3.8' python_version: '3.8'
- cmake_windows_gpu: - cmake_windows_gpu:
cu_version: cu102 cu_version: cu113
name: cmake_windows_gpu name: cmake_windows_gpu
python_version: '3.8' python_version: '3.8'
- cmake_macos_cpu: - cmake_macos_cpu:
......
...@@ -268,9 +268,9 @@ def cmake_workflows(indentation=6): ...@@ -268,9 +268,9 @@ def cmake_workflows(indentation=6):
for device in device_types: for device in device_types:
job = {"name": f"cmake_{os_type}_{device}", "python_version": python_version} 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": 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}) jobs.append({f"cmake_{os_type}_{device}": job})
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