Skip to content

Commit ac19e81

Browse files
committed
Add GPU detection and enable workflow on feature branch
- Add host GPU hardware detection with lspci and nvidia-smi - Add container GPU access test with --gpus all flag - Enable workflow triggers on cd/initial-ci branch for testing - Will show RTX Pro 6000 GPU information when running on GPU runners Signed-off-by: cdunning <cdunning@nvidia.com>
1 parent 5b6e950 commit ac19e81

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

.github/workflows/ci.yml

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,12 @@ jobs:
2121
docker --version
2222
docker run --rm hello-world
2323
24+
- name: Check GPU Hardware
25+
run: |
26+
echo "=== Host GPU Information ==="
27+
lspci | grep -i nvidia || echo "No NVIDIA GPU found via lspci"
28+
nvidia-smi || echo "nvidia-smi not available on host"
29+
2430
- name: Test CUDA Image
2531
run: |
2632
docker pull nvidia/cuda:13.0.2-base-ubuntu24.04
@@ -29,3 +35,13 @@ jobs:
2935
cat /etc/os-release
3036
which nvcc || echo 'nvcc not in PATH'
3137
"
38+
39+
- name: Test CUDA Image with GPU Access
40+
run: |
41+
echo "=== Testing GPU access in container ==="
42+
docker run --rm --gpus all nvidia/cuda:13.0.2-base-ubuntu24.04 bash -c "
43+
echo 'Container with GPU access:'
44+
nvidia-smi || echo 'nvidia-smi not available in container'
45+
echo 'CUDA Runtime version:'
46+
cat /usr/local/cuda/version.txt 2>/dev/null || echo 'CUDA version file not found'
47+
" || echo "GPU access failed - may not be available on this runner"

0 commit comments

Comments
 (0)