From 89ab903fe699e3aafa9906a3f7a261ffdcbc2000 Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 12 Jun 2023 07:28:51 -0700 Subject: [PATCH] Add nvidia-nvjitlink-cu12 to s3 manage --- s3_management/manage.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/s3_management/manage.py b/s3_management/manage.py index 8a3c9dcf9..655f7de40 100644 --- a/s3_management/manage.py +++ b/s3_management/manage.py @@ -72,7 +72,7 @@ "nvidia_cusparse_cu12", "nvidia_nccl_cu12", "nvidia_nvtx_cu12", - "nvidia-nvjitlink-cu12", + "nvidia_nvjitlink_cu12", "packaging", "portalocker", "pytorch_triton",