Skip to content

Commit 20e3582

Browse files
committed
chore(ci): modprobe nvidia for gpu tests
1 parent 1913415 commit 20e3582

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

.github/workflows/ci.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,11 @@ jobs:
163163
sudo apt-get update
164164
sudo apt-get install --no-install-recommends --yes cuda-drivers
165165
rm nvidia-driver-local-repo-ubuntu2404-570.148.08_1.0-1_amd64.deb
166+
- name: Load NVIDIA kernel modules and verify GPU
167+
run: |
168+
sudo modprobe nvidia
169+
sudo modprobe nvidia-uvm
170+
nvidia-smi || echo "WARNING: nvidia-smi failed, GPU may not be available"
166171
- name: Install required packages
167172
run: |
168173
sudo apt-get update

0 commit comments

Comments
 (0)