Skip to content

Commit 0f9bc16

Browse files
authored
chore(ci) Use CUDA network installer and specify sub-packages. (#189)
The network installer is a smaller initial download, which will only download the require packages to install. This should reduce the install times on Windows CI.
1 parent b85d9ca commit 0f9bc16

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

.github/workflows/rust.yml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,18 +26,22 @@ jobs:
2626
target: x86_64-unknown-linux-gnu
2727
cuda: "11.2.2"
2828
linux-local-args: ["--toolkit"]
29+
sub-packages: []
2930
- os: ubuntu-24.04
3031
target: x86_64-unknown-linux-gnu
3132
cuda: "12.8.1"
3233
linux-local-args: ["--toolkit"]
34+
sub-packages: []
3335
- os: windows-latest
3436
target: x86_64-pc-windows-msvc
3537
cuda: "11.2.2"
3638
linux-local-args: []
39+
sub-packages: ["nvcc", "nvrtc", "nvrtc_dev", "cudart", "cublas", "cublas_dev", "curand", "curand_dev"]
3740
- os: windows-latest
3841
target: x86_64-pc-windows-msvc
3942
cuda: "12.8.1"
4043
linux-local-args: []
44+
sub-packages: ["nvcc", "nvrtc", "nvrtc_dev", "cuda_profiler_api", "cudart", "cublas", "cublas_dev", "curand", "curand_dev"]
4145

4246
steps:
4347
- name: Checkout repository
@@ -48,8 +52,11 @@ jobs:
4852
id: cuda-toolkit
4953
with:
5054
cuda: ${{ matrix.cuda }}
55+
method: network
5156
linux-local-args: ${{ toJson(matrix.linux-local-args) }}
5257
use-local-cache: false
58+
sub-packages: ${{ toJson(matrix.sub-packages) }}
59+
log-file-suffix: '${{matrix.os}}-${{matrix.cuda}}'
5360

5461
- name: Verify CUDA installation
5562
run: nvcc --version

0 commit comments

Comments
 (0)