From 4a5a66cbaff2c7e4ecad17249c29ec245876de5d Mon Sep 17 00:00:00 2001 From: Ryo Yamashita Date: Mon, 22 Jul 2024 15:59:46 +0900 Subject: [PATCH] CUDA v12.4.1 --- .github/workflows/build.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 7ef0b0e..04b19f9 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -41,7 +41,7 @@ jobs: release_config: Release - artifact_name: onnxruntime-win-x64-gpu os: windows-2022 - cuda_version: 12.5.0 + cuda_version: 12.4.1 # Windowsの場合デフォルトのパッケージ群では不十分であるため、必要そうなパッケージを指定する。ただしいくつかは不要かもしれない cuda_sub_packages: '["cudart", "cuobjdump", "nvcc", "nvdisasm", "thrust", "cublas_dev", "cufft_dev", "curand_dev", "cusolver_dev", "cusparse_dev", "visual_studio_integration"]' cudnn_url: https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/windows-x86_64/cudnn-windows-x86_64-8.9.7.29_cuda12-archive.zip @@ -60,7 +60,7 @@ jobs: release_config: Release - artifact_name: onnxruntime-linux-x64-gpu os: ubuntu-20.04 - cuda_version: 12.5.0 + cuda_version: 12.4.1 cuda_sub_packages: "[]" # デフォルト cudnn_url: https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz build_opts: --cmake_extra_defines CMAKE_SYSTEM_NAME=Linux CMAKE_SYSTEM_PROCESSOR=x86_64 --config Release --parallel --compile_no_warning_as_error --update --build --build_shared_lib --use_cuda