Skip to content

Commit

Permalink
build: pin cuda requirements to latest supported versions
Browse files Browse the repository at this point in the history
  • Loading branch information
andreabak committed Sep 13, 2024
1 parent 941b6c4 commit 0ab6459
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions requirements-cuda.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
nvidia-cublas-cu12
nvidia-cudnn-cu12
nvidia-cublas-cu12~=12.0
nvidia-cudnn-cu12~=8.0

0 comments on commit 0ab6459

Please sign in to comment.