"tests/models/multimodal/pooling/test_colpali.py" did not exist on "d1ea65d0a1c606ae041b73fd45ccd33980ca08e7"
gpu.md 3.26 KB
Newer Older
1
2
3
4
---
toc_depth: 3
---

5
6
7
8
9
10
# GPU

vLLM is a Python library that supports the following GPU variants. Select your GPU type to see vendor specific instructions:

=== "NVIDIA CUDA"

11
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:installation"
12
13
14

=== "AMD ROCm"

15
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:installation"
16
17
18

=== "Intel XPU"

19
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:installation"
20
21
22
23

## Requirements

- OS: Linux
24
- Python: 3.10 -- 3.13
25

26
27
28
!!! note
    vLLM does not support Windows natively. To run vLLM on Windows, you can use the Windows Subsystem for Linux (WSL) with a compatible Linux distribution, or use some community-maintained forks, e.g. [https://github.com/SystemPanic/vllm-windows](https://github.com/SystemPanic/vllm-windows).

29
30
=== "NVIDIA CUDA"

31
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:requirements"
32
33
34

=== "AMD ROCm"

35
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:requirements"
36
37
38

=== "Intel XPU"

39
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:requirements"
40
41
42
43
44
45
46
47
48

## Set up using Python

### Create a new Python environment

--8<-- "docs/getting_started/installation/python_env_setup.inc.md"

=== "NVIDIA CUDA"

49
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:set-up-using-python"
50
51
52

=== "AMD ROCm"

53
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:set-up-using-python"
54
55
56

=== "Intel XPU"

57
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:set-up-using-python"
58

59
### Pre-built wheels {#pre-built-wheels}
60
61
62

=== "NVIDIA CUDA"

63
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:pre-built-wheels"
64
65
66

=== "AMD ROCm"

67
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:pre-built-wheels"
68
69
70

=== "Intel XPU"

71
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:pre-built-wheels"
72
73
74
75
76

### Build wheel from source

=== "NVIDIA CUDA"

77
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:build-wheel-from-source"
78
79
80

=== "AMD ROCm"

81
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:build-wheel-from-source"
82
83
84

=== "Intel XPU"

85
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:build-wheel-from-source"
86
87
88
89
90

## Set up using Docker

### Pre-built images

91
--8<-- [start:pre-built-images]
92

93
94
=== "NVIDIA CUDA"

95
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:pre-built-images"
96
97
98

=== "AMD ROCm"

99
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:pre-built-images"
100
101
102

=== "Intel XPU"

103
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:pre-built-images"
104

105
--8<-- [end:pre-built-images]
106

107
### Build image from source
108

109
--8<-- [start:build-image-from-source]
110
111
112

=== "NVIDIA CUDA"

113
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:build-image-from-source"
114
115
116

=== "AMD ROCm"

117
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:build-image-from-source"
118
119
120

=== "Intel XPU"

121
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:build-image-from-source"
122

123
--8<-- [end:build-image-from-source]
124

125
126
127
128
## Supported features

=== "NVIDIA CUDA"

129
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:supported-features"
130
131
132

=== "AMD ROCm"

133
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:supported-features"
134
135
136

=== "Intel XPU"

137
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:supported-features"