diff --git a/s3_management/manage.py b/s3_management/manage.py index cb5783f38..8a3c9dcf9 100644 --- a/s3_management/manage.py +++ b/s3_management/manage.py @@ -72,6 +72,7 @@ "nvidia_cusparse_cu12", "nvidia_nccl_cu12", "nvidia_nvtx_cu12", + "nvidia-nvjitlink-cu12", "packaging", "portalocker", "pytorch_triton",