diff --git a/training/requirements.txt b/training/requirements.txt index 75f6a5b..00a5c53 100644 --- a/training/requirements.txt +++ b/training/requirements.txt @@ -47,7 +47,7 @@ numpy==1.26.3 nvidia-cublas-cu11==11.10.3.66 nvidia-cuda-cupti-cu11==11.7.101 nvidia-cuda-nvrtc-cu11==11.7.99 -nvidia-cuda-runtime-cu11==11.7.99 +nvidia-cuda-runtime-cu11==11.8.89 nvidia-cudnn-cu11==8.5.0.96 nvidia-cufft-cu11==10.9.0.58 nvidia-curand-cu11==10.2.10.91