Skip to content

Commit ac7dc76

Browse files
authored
misc: Update CI docker container to use latest cudnn (#1362)
<!-- .github/pull_request_template.md --> ## 📌 Description Add nvidia-cudnn-cu12 and nvidia-cudnn-frontend to the docker container. ## 🔍 Related Issues #1283 #1312 #1317 ## 🚀 Pull Request Checklist Thank you for contributing to FlashInfer! Before we review your pull request, please make sure the following items are complete. ### ✅ Pre-commit Checks - [ ] I have installed `pre-commit` by running `pip install pre-commit` (or used your preferred method). - [ ] I have installed the hooks with `pre-commit install`. - [ ] I have run the hooks manually with `pre-commit run --all-files` and fixed any reported issues. > If you are unsure about how to set up `pre-commit`, see [the pre-commit documentation](https://pre-commit.com/). ## 🧪 Tests - [ ] Tests have been added or updated as needed. - [ ] All tests are passing (`unittest`, etc.). ## Reviewer Notes cc @Anerudhan
1 parent c261e97 commit ac7dc76

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

docker/install/install_python_packages.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,3 +21,4 @@ set -u
2121

2222
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
2323
pip3 install ninja pytest numpy scipy build pynvml cuda-python einops nvidia-nvshmem-cu12
24+
pip3 install 'nvidia-cudnn-frontend>=1.13.0' 'nvidia-cudnn-cu12>=9.11.0.98'

0 commit comments

Comments
 (0)