Skip to content

Commit 6610856

Browse files
zewenli98zewenli98
committed
add pycuda package for tests
1 parent 9df10a4 commit 6610856

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

‎tests/modules/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
11
timm==v0.9.2
22
transformers==4.30.0
33
torchvision
4+
pycuda

0 commit comments

Comments
 (0)