Skip to content

Commit b8d3ec6

Browse files
committed
Install CUDA toolkit on GPU runner for CI
The ubuntu-gpu-t4-4-core runner has the GPU driver but not the CUDA toolkit (nvcc). Add Jimver/cuda-toolkit action to install CUDA 12.6.3 with nvcc, cudart-dev, cublas-dev, and curand-dev.
1 parent e309462 commit b8d3ec6

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

.github/workflows/ci_cuda.yaml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,12 @@ jobs:
2323
steps:
2424
- name: Checkout repository
2525
uses: actions/checkout@v6
26+
- name: Install CUDA toolkit
27+
uses: Jimver/cuda-toolkit@v0.2.21
28+
with:
29+
cuda: '12.6.3'
30+
method: 'network'
31+
sub-packages: '["nvcc", "cudart-dev", "cublas-dev", "curand-dev"]'
2632
- name: Install dependencies
2733
run: sudo apt-get update && sudo apt-get install -y curl build-essential libssl-dev protobuf-compiler pkg-config
2834
- name: Install Rust Stable

0 commit comments

Comments
 (0)