Skip to content

Commit 28f032d

Browse files
committed
ci: install only cublas
1 parent 2f5f29c commit 28f032d

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

.github/workflows/test-build.yml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,13 @@ jobs:
2727
cuda: "12.9.1"
2828
linux-local-args: '["--toolkit"]'
2929
method: "network"
30+
sub-packages: '["nvcc"]'
31+
32+
- name: Install cusparse
33+
run: sudo apt-get install -y libcusparse-dev-12-9
34+
35+
- name: Install cublas
36+
run: sudo apt-get install -y libcublas-dev-12-9
3037

3138
- name: Install PyTorch & Build Dependencies
3239
run: |

0 commit comments

Comments
 (0)