Skip to content

Commit 13164b4

Browse files
authored
Add cu12 packages to allow list for download.pytorch.org (#1420)
1 parent 1a3174e commit 13164b4

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

s3_management/manage.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,17 @@
6161
"nvidia_cusparse_cu11",
6262
"nvidia_nccl_cu11",
6363
"nvidia_nvtx_cu11",
64+
"nvidia_cublas_cu12",
65+
"nvidia_cuda_cupti_cu12",
66+
"nvidia_cuda_nvrtc_cu12",
67+
"nvidia_cuda_runtime_cu12",
68+
"nvidia_cudnn_cu12",
69+
"nvidia_cufft_cu12",
70+
"nvidia_curand_cu12",
71+
"nvidia_cusolver_cu12",
72+
"nvidia_cusparse_cu12",
73+
"nvidia_nccl_cu12",
74+
"nvidia_nvtx_cu12",
6475
"packaging",
6576
"portalocker",
6677
"pytorch_triton",

0 commit comments

Comments
 (0)