From 5727a84ca630a46f22c4a32a6448573cc68d2823 Mon Sep 17 00:00:00 2001 From: silverwolf-x <185666132@qq.com> Date: Sat, 23 Nov 2024 16:36:55 +0800 Subject: [PATCH] chore: cuda --- .github/workflows/lightgbm.yml | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/.github/workflows/lightgbm.yml b/.github/workflows/lightgbm.yml index 276dd03..38a7707 100644 --- a/.github/workflows/lightgbm.yml +++ b/.github/workflows/lightgbm.yml @@ -10,7 +10,6 @@ on: jobs: build: runs-on: windows-2022 - steps: - uses: actions/setup-python@v5 with: @@ -30,18 +29,18 @@ jobs: id: cuda-toolkit # https://docs.nvidia.com/cuda/archive/12.6.2/cuda-installation-guide-microsoft-windows/index.html#id3 with: - cuda: '12.1.1' - method: 'network' + cuda: 12.1.1 + method: local sub-packages: '["nvcc", "visual_studio_integration"]' # Disable caching of the CUDA binaries, since it does not give us any significant performance improvement - use-github-cache: "false" + uuse-local-cache: false # 在GITHUB_ENV中输出CUDA_PATH - name: echo version run: | nvcc -V echo ${{ steps.cuda-toolkit.outputs.cuda }} - echo ${{ env.CUDA_PATH }} + echo "${{ env.CUDA_PATH }}" echo ${{ steps.install-boost.outputs.BOOST_VER }} - name: Configure build run: | @@ -51,10 +50,7 @@ jobs: python -V # pip wheel lightgbm --config-settings=cmake.define.Boost_ROOT=$BOOST_ROOT - pip wheel lightgbm \ - --config-settings=cmake.define.USE_GPU=ON \ - --config-settings=cmake.define.OpenCL_INCLUDE_DIR=$CUDA_PATH/include \ - --config-settings=cmake.define.Boost_ROOT=$BOOST_ROOT + pip wheel lightgbm --config-settings=cmake.define.USE_GPU=ON --config-settings=cmake.define.OpenCL_INCLUDE_DIR="$CUDA_PATH/include"--config-settings=cmake.define.Boost_ROOT=$BOOST_ROOT env: BOOST_ROOT: ${{ steps.install-boost.outputs.BOOST_ROOT }}