alihan/light-the-torch
1
0
Fork 0
mirror of https://github.com/pmeier/light-the-torch.git synced 2024-09-08 23:29:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

path more 3rd party packages (#137)

Browse Source
This commit is contained in:
Philip Meier
2023-05-16 09:54:53 +02:00
committed by GitHub
parent f9b67a8691
commit 20a75752e1
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
light_the_torch/_patch.py
View File

@@ -57,11 +57,23 @@ THIRD_PARTY_PACKAGES = {
"cmake",
"colorama",
"filelock",
"fsspec",
"idna",
"lit",
"mpmath",
"networkx",
"numpy",
"nvidia-cublas-cu11",
"nvidia-cuda-cupti-cu11",
"nvidia-cuda-nvrtc-cu11",
"nvidia-cuda-runtime-cu11",
"nvidia-cudnn-cu11",
"nvidia-cufft-cu11",
"nvidia-curand-cu11",
"nvidia-cusolver-cu11",
"nvidia-cusparse-cu11",
"nvidia-nccl-cu11",
"nvidia-nvtx-cu11",
"packaging",
"portalocker",
"requests",