[Docs] Update installation instructions to include CUDA 11.8 xFormers (…

…vllm-project#2246)
madsys-dev · Dec 23, 2023 · 1db83e3 · 1db83e3
1 parent a1b9cb2
commit 1db83e3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/source/getting_started/installation.rst b/docs/source/getting_started/installation.rst
@@ -42,6 +42,10 @@ You can install vLLM using pip:
         $ pip uninstall torch -y
         $ pip install torch --upgrade --index-url https://download.pytorch.org/whl/cu118
 
+        $ # Re-install xFormers with CUDA 11.8.
+        $ pip uninstall xformers -y
+        $ pip install --upgrade xformers --index-url https://download.pytorch.org/whl/cu118
+
 
 .. _build_from_source: