diff --git a/setup.py b/setup.py index 0fdabd758b..edcf310854 100644 --- a/setup.py +++ b/setup.py @@ -64,7 +64,7 @@ else: requirements_cuda = [ "nvidia-cuda-runtime-cu12>=12.1.105,<12.6", - "nvidia-nvjitlink-cu12<12.6.85", + "nvidia-nvjitlink-cu12<12.6", "nvidia-cublas-cu12>=12.1.3.1", "nvidia-cusolver-cu12>=11.4.5.107", "nvidia-cusparse-cu12>=12.1.0.106,<12.5.4.2",