diff --git a/s3_management/manage.py b/s3_management/manage.py index 757ac0833..cb5783f38 100644 --- 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",