Add cu12 packages to allow list for download.pytorch.org (#1420)

pytorch · Jun 9, 2023 · 13164b4 · 13164b4
1 parent 1a3174e
commit 13164b4
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/s3_management/manage.py b/s3_management/manage.py
@@ -61,6 +61,17 @@
     "nvidia_cusparse_cu11",
     "nvidia_nccl_cu11",
     "nvidia_nvtx_cu11",
+    "nvidia_cublas_cu12",
+    "nvidia_cuda_cupti_cu12",
+    "nvidia_cuda_nvrtc_cu12",
+    "nvidia_cuda_runtime_cu12",
+    "nvidia_cudnn_cu12",
+    "nvidia_cufft_cu12",
+    "nvidia_curand_cu12",
+    "nvidia_cusolver_cu12",
+    "nvidia_cusparse_cu12",
+    "nvidia_nccl_cu12",
+    "nvidia_nvtx_cu12",
     "packaging",
     "portalocker",
     "pytorch_triton",