diff --git a/requirements.txt b/requirements.txt index 4e5c305..9451430 100644 --- a/requirements.txt +++ b/requirements.txt @@ -48,7 +48,7 @@ nvidia-curand-cu12==10.3.7.37 nvidia-cusolver-cu12==11.4.5.107 nvidia-cusparse-cu12==12.1.0.106 nvidia-nccl-cu12==2.20.5 -nvidia-nvjitlink-cu12==12.5.40 +nvidia-nvjitlink-cu12==12.6.77 nvidia-nvtx-cu12==12.1.105 omegaconf==2.3.0 optuna==2.10.1