diff --git a/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl b/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl index 6934b75b4..417a23705 100644 --- a/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl +++ b/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl @@ -58,6 +58,10 @@ CUDA_REDIST_JSON_DICT = { "https://developer.download.nvidia.com/compute/cuda/redist/redistrib_12.6.0.json", "87740b01676b3d18982982ab96ec7fa1a626d03a96df070a6b0f258d01ff5fab", ], + "12.6.1": [ + "https://developer.download.nvidia.com/compute/cuda/redist/redistrib_12.6.1.json", + "22ddfeb81a6f9cee4a708a2e3b4db1c36c7db0a1daa1f33f9c7f2f12a1e790de", + ], } CUDNN_REDIST_JSON_DICT = { @@ -134,6 +138,7 @@ CUDA_NCCL_WHEELS = { "12.5.0": CUDA_12_NCCL_WHEEL_DICT, "12.5.1": CUDA_12_NCCL_WHEEL_DICT, "12.6.0": CUDA_12_NCCL_WHEEL_DICT, + "12.6.1": CUDA_12_NCCL_WHEEL_DICT, } REDIST_VERSIONS_TO_BUILD_TEMPLATES = {