Skip to content

GitLab

Unverified Commit 82b110d5 authored by Kevin H. Luu's avatar Kevin H. Luu Committed by GitHub
Browse files

[ci] Bound nvidia-cudnn-frontend version (#36719) 


Signed-off-by: default avatarkhluu <khluu000@gmail.com>
parent 9040cd40
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
requirements/cuda.txt
View file @ 82b110d5
...@@ -10,6 +10,9 @@ torchaudio==2.10.0 ...@@ -10,6 +10,9 @@ torchaudio==2.10.0
torchvision==0.25.0 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version torchvision==0.25.0 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
# FlashInfer should be updated together with the Dockerfile # FlashInfer should be updated together with the Dockerfile
flashinfer-python==0.6.4 flashinfer-python==0.6.4
# Cap nvidia-cudnn-frontend (transitive dep of flashinfer) due to
# breaking changes in 1.19.0
nvidia-cudnn-frontend>=1.13.0,<1.19.0
# QuACK and Cutlass DSL for FA4 (cute-DSL implementation) # QuACK and Cutlass DSL for FA4 (cute-DSL implementation)
nvidia-cutlass-dsl>=4.4.0.dev1 nvidia-cutlass-dsl>=4.4.0.dev1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment