From ac908b38284a6747a27a23adf1e0d2ac665ff536 Mon Sep 17 00:00:00 2001 From: atalman Date: Mon, 12 Jun 2023 06:42:44 -0700 Subject: [PATCH] Add nvidia-nvjitlink-cu12 to s3 manage --- s3_management/manage.py | 1 + 1 file changed, 1 insertion(+) 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",