From 9204e31dc45a53ad58fc8605f85ad88a4521de06 Mon Sep 17 00:00:00 2001 From: Ryo Yamashita Date: Thu, 15 Aug 2024 22:16:00 +0900 Subject: [PATCH] =?UTF-8?q?remove:=20CUDA=E3=81=A8DirectML=E3=81=AE?= =?UTF-8?q?=E4=B8=A1=E6=96=B9=E3=82=92=E6=9C=89=E5=8A=B9=E5=8C=96=E3=81=97?= =?UTF-8?q?=E3=81=9F=E3=83=93=E3=83=AB=E3=83=89=E3=81=AF=E6=8F=90=E4=BE=9B?= =?UTF-8?q?=E3=81=97=E3=81=AA=E3=81=84=20(#47)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .github/workflows/build.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index e702480..1f535af 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -37,18 +37,18 @@ jobs: build_opts: --cmake_extra_defines CMAKE_SYSTEM_NAME=Windows CMAKE_SYSTEM_PROCESSOR=x86_64 --config Release --parallel --compile_no_warning_as_error --update --build --build_shared_lib result_dir: build/Release release_config: Release - - artifact_name: onnxruntime-win-x64-gpu-cuda + - artifact_name: onnxruntime-win-x64-dml os: windows-2022 build_opts: --cmake_extra_defines CMAKE_SYSTEM_NAME=Windows CMAKE_SYSTEM_PROCESSOR=x86_64 --config Release --parallel --compile_no_warning_as_error --update --build --build_shared_lib --use_dml result_dir: build/Release release_config: Release - - artifact_name: onnxruntime-win-x64-gpu + - artifact_name: onnxruntime-win-x64-cuda os: windows-2022 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 - build_opts: --cmake_extra_defines CMAKE_SYSTEM_NAME=Windows CMAKE_SYSTEM_PROCESSOR=x86_64 --config Release --parallel --compile_no_warning_as_error --update --build --build_shared_lib --use_dml --use_cuda --cuda_version 12.4 + build_opts: --cmake_extra_defines CMAKE_SYSTEM_NAME=Windows CMAKE_SYSTEM_PROCESSOR=x86_64 --config Release --parallel --compile_no_warning_as_error --update --build --build_shared_lib --use_cuda --cuda_version 12.4 result_dir: build/Release release_config: Release - artifact_name: onnxruntime-win-x86 @@ -61,7 +61,7 @@ jobs: 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 result_dir: build release_config: Release - - artifact_name: onnxruntime-linux-x64-gpu + - artifact_name: onnxruntime-linux-x64-cuda os: ubuntu-20.04 cuda_version: 12.4.1 cuda_sub_packages: "[]" # デフォルト