Skip to content

Commit c3ff7e7

Browse files
authored
Ensure docker installs the torch version we need (#1901)
<!-- .github/pull_request_template.md --> ## 📌 Description adds --force-reinstall when installing torch. This ensures we install the version we want, even if requirements.txt installed another one. ## 🔍 Related Issues <!-- Link any related issues here --> ## 🚀 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). - [x] 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 <!-- Optional: anything you'd like reviewers to focus on, concerns, etc. -->
1 parent 40d3fea commit c3ff7e7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docker/install/install_python_packages.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ CUDA_VERSION=${1:-cu128}
2424

2525
# Install requirements.txt first, so we can override any versions later for docker image
2626
pip3 install -r /install/requirements.txt
27-
pip3 install torch --index-url https://download.pytorch.org/whl/${CUDA_VERSION}
27+
pip3 install --force-reinstall torch --index-url https://download.pytorch.org/whl/${CUDA_VERSION}
2828
pip3 install responses pytest scipy build cuda-python nvidia-nvshmem-cu12
2929

3030
# Install cudnn package based on CUDA version

0 commit comments

Comments
 (0)