diff --git a/.circleci/config.yml b/.circleci/config.yml index 3a58071fabf..27b2fff2efe 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -1642,16 +1642,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: diff --git a/.circleci/regenerate.py b/.circleci/regenerate.py index ac903fd0f71..2413e4adbac 100755 --- a/.circleci/regenerate.py +++ b/.circleci/regenerate.py @@ -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) diff --git a/packaging/windows/internal/cuda_install.bat b/packaging/windows/internal/cuda_install.bat index db181c62321..0b42a2cbba3 100644 --- a/packaging/windows/internal/cuda_install.bat +++ b/packaging/windows/internal/cuda_install.bat @@ -167,7 +167,7 @@ if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" ( curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" if errorlevel 1 exit /b 1 set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" - set "ARGS=thrust_11.3 nvcc_11.3 cuobjdump_11.3 nvprune_11.3 nvprof_11.3 cupti_11.3 cublas_11.3 cublas_dev_11.3 cudart_11.3 cufft_11.3 cufft_dev_11.3 curand_11.3 curand_dev_11.3 cusolver_11.3 cusolver_dev_11.3 cusparse_11.3 cusparse_dev_11.3 npp_11.3 npp_dev_11.3 nvrtc_11.3 nvrtc_dev_11.3 nvml_dev_11.3" + set "ARGS=thrust_11.3 nvcc_11.3 cuobjdump_11.3 nvprune_11.3 nvprof_11.3 cupti_11.3 cublas_11.3 cublas_dev_11.3 cudart_11.3 cufft_11.3 cufft_dev_11.3 curand_11.3 curand_dev_11.3 cusolver_11.3 cusolver_dev_11.3 cusparse_11.3 cusparse_dev_11.3 npp_11.3 npp_dev_11.3 nvjpeg_11.3 nvjpeg_dev_11.3 nvrtc_11.3 nvrtc_dev_11.3 nvml_dev_11.3" )