diff --git a/server/pyproject.toml b/server/pyproject.toml index 2c60b982..010b349c 100644 --- a/server/pyproject.toml +++ b/server/pyproject.toml @@ -51,8 +51,8 @@ grpcio-tools = "^1.51.1" pytest = "^7.3.0" [tool.poetry.group.cuda.dependencies] -# Fix https://github.com/NVIDIA/nccl/issues/1251 bug in nccl 2.20.5 bundled with PyTorch 2.3, nccl 2.21 appears to fix the issue. -nvidia-nccl-cu12 = "==2.21.5" +# Fix https://github.com/NVIDIA/nccl/issues/1251 bug in nccl 2.20.5 bundled with PyTorch 2.3, nccl 2.22.3 appears to fix the issue. +nvidia-nccl-cu12 = "==2.22.3" [[tool.poetry.source]] name = "pytorch-gpu-src"