Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tianlh
LightGBM-DCU
Commits
dcb5d973
Unverified
Commit
dcb5d973
authored
Jul 28, 2025
by
Maik Riechert
Committed by
GitHub
Jul 28, 2025
Browse files
[ci] [CUDA] Switch to GitHub runner for GPU CI (#6958)
parent
8478020e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
46 deletions
+3
-46
.github/workflows/cuda.yml
.github/workflows/cuda.yml
+3
-46
No files found.
.github/workflows/cuda.yml
View file @
dcb5d973
...
...
@@ -9,12 +9,6 @@ on:
-
master
# Run manually by clicking a button in the UI
workflow_dispatch
:
inputs
:
restart_docker
:
description
:
'
Restart
nvidia-docker
on
the
runner
before
building?'
required
:
true
type
:
boolean
default
:
false
# automatically cancel in-progress builds if another commit is pushed
concurrency
:
...
...
@@ -22,49 +16,12 @@ concurrency:
cancel-in-progress
:
true
jobs
:
# Optionally reinstall + restart docker on the runner before building.
# This is safe as long as only 1 of these jobs runs at a time.
restart-docker
:
name
:
set up docker
runs-on
:
[
self-hosted
,
linux
]
timeout-minutes
:
30
steps
:
-
name
:
Setup or update software on host machine
if
:
${{ inputs.restart_docker }}
run
:
|
# install core packages
sudo apt-get update
sudo apt-get install --no-install-recommends -y \
apt-transport-https \
ca-certificates \
curl \
gnupg-agent \
lsb-release \
software-properties-common
# set up nvidia-docker
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
sudo add-apt-repository -y \
"deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"
curl -sL https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -sL \
https://nvidia.github.io/nvidia-docker/$(. /etc/os-release;echo $ID$VERSION_ID)/nvidia-docker.list \
| sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install --no-install-recommends -y \
containerd.io \
docker-ce \
docker-ce-cli \
nvidia-docker2
sudo chmod a+rw /var/run/docker.sock
sudo systemctl restart docker
-
name
:
mark job successful
run
:
|
exit 0
test
:
# yamllint disable-line rule:line-length
name
:
${{ matrix.task }} ${{ matrix.cuda_version }} ${{ matrix.method }} (${{ matrix.linux_version }}, ${{ matrix.compiler }}, Python ${{ matrix.python_version }})
runs-on
:
[
self-hosted
,
linux
]
needs
:
[
restart-docker
]
# yamllint disable-line rule:line-length
# ref: https://docs.github.com/en/actions/concepts/runners/about-larger-runners#specifications-for-gpu-larger-runners
runs-on
:
LightGBM-GPU
container
:
image
:
nvcr.io/nvidia/cuda:${{ matrix.cuda_version }}-devel-${{ matrix.linux_version }}
env
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment