Skip to content

Commit

Permalink
chore: cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
Silverwolf-x committed Nov 23, 2024
1 parent 9272c61 commit 5727a84
Showing 1 changed file with 5 additions and 9 deletions.
14 changes: 5 additions & 9 deletions .github/workflows/lightgbm.yml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@ on:
jobs:
build:
runs-on: windows-2022

steps:
- uses: actions/setup-python@v5
with:
Expand All @@ -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: |
Expand All @@ -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 }}

Expand Down

0 comments on commit 5727a84

Please sign in to comment.