diff --git a/.github/workflows/cuda.yml b/.github/workflows/cuda.yml index 6c32db1a2ed..f6bb354bc5a 100644 --- a/.github/workflows/cuda.yml +++ b/.github/workflows/cuda.yml @@ -82,13 +82,13 @@ jobs: - method: wheel compiler: gcc python_version: "3.10" - cuda_version: "11.8.0" - linux_version: "ubuntu20.04" + cuda_version: "12.6.1" + linux_version: "ubuntu22.04" task: cuda - method: source compiler: gcc python_version: "3.12" - cuda_version: "12.2.0" + cuda_version: "12.2.2" linux_version: "ubuntu22.04" task: cuda - method: pip