Skip to content

Commit 19ebc40

Browse files
committed
Use tag
Signed-off-by: Huy Do <[email protected]>
1 parent 5b350ae commit 19ebc40

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

docs/ci/update_pytorch_version.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,20 +94,20 @@ Here is how to build and install it from source with torch2.7.0+cu128 as done in
9494
```
9595
export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.9 9.0 10.0+PTX'
9696
export FLASHINFER_ENABLE_SM90=1
97-
uv pip install --system --no-build-isolation "git+https://github.com/flashinfer-ai/flashinfer@21ea1d2545f74782b91eb8c08fd503ac4c0743fc"
97+
uv pip install --system --no-build-isolation "git+https://github.com/flashinfer-ai/flashinfer@v0.2.6.post1"
9898
```
9999

100100
A small gotcha here is that building FlashInfer from source adds around 30
101101
minutes to vLLM build time. Thus, it’s better to cache the wheel somewhere
102-
public so that it can be installed right away, for example https://download.pytorch.org/whl/cu128/flashinfer/flashinfer_python-0.2.5%2Bcu128torch2.7-cp38-abi3-linux_x86_64.whl. In future releases, please reach out to PyTorch release
102+
public so that it can be installed right away, for example https://download.pytorch.org/whl/cu128/flashinfer/flashinfer_python-0.2.6.post1%2Bcu128torch2.7-cp39-abi3-linux_x86_64.whl. In future releases, please reach out to PyTorch release
103103
team if you want to get the package published there.
104104

105105
### xFormers
106106
Similar to FlashInfer, here is how to build and install xFormers from source:
107107

108108
```
109-
export TORCH_CUDA_ARCH_LIST='7.5 8.0 8.6 8.9 9.0 10.0+PTX'
110-
MAX_JOBS=16 uv pip install --system --no-build-isolation "git+https://github.com/facebookresearch/xformers@f2de641ef670510cadab099ce6954031f52f191c"
109+
export TORCH_CUDA_ARCH_LIST='7.0 7.5 8.0 8.9 9.0 10.0+PTX'
110+
MAX_JOBS=16 uv pip install --system --no-build-isolation "git+https://github.com/facebookresearch/xformers@v0.0.30"
111111
```
112112

113113
### Mamba

0 commit comments

Comments
 (0)