Skip to content

Commit 20a7575

Browse files
authored
path more 3rd party packages (#137)
1 parent f9b67a8 commit 20a7575

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

light_the_torch/_patch.py

+12
Original file line numberDiff line numberDiff line change
@@ -57,11 +57,23 @@ def from_str(cls, string):
5757
"cmake",
5858
"colorama",
5959
"filelock",
60+
"fsspec",
6061
"idna",
6162
"lit",
6263
"mpmath",
6364
"networkx",
6465
"numpy",
66+
"nvidia-cublas-cu11",
67+
"nvidia-cuda-cupti-cu11",
68+
"nvidia-cuda-nvrtc-cu11",
69+
"nvidia-cuda-runtime-cu11",
70+
"nvidia-cudnn-cu11",
71+
"nvidia-cufft-cu11",
72+
"nvidia-curand-cu11",
73+
"nvidia-cusolver-cu11",
74+
"nvidia-cusparse-cu11",
75+
"nvidia-nccl-cu11",
76+
"nvidia-nvtx-cu11",
6577
"packaging",
6678
"portalocker",
6779
"requests",

0 commit comments

Comments
 (0)