From 0ab645914ffb114e747823c2c1d0a2e620cb8b3e Mon Sep 17 00:00:00 2001 From: abk16 Date: Fri, 13 Sep 2024 18:55:25 +0200 Subject: [PATCH] build: pin cuda requirements to latest supported versions --- requirements-cuda.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements-cuda.txt b/requirements-cuda.txt index a783e05..090c804 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -1,2 +1,2 @@ -nvidia-cublas-cu12 -nvidia-cudnn-cu12 +nvidia-cublas-cu12~=12.0 +nvidia-cudnn-cu12~=8.0