Add nvidia-nvjitlink-cu12 to s3 manage

pytorch · Jun 12, 2023 · ac908b3 · ac908b3
1 parent d7f499d
commit ac908b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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",