gpu.md 3.1 KB
Newer Older
1
2
3
4
5
6
# GPU

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

=== "NVIDIA CUDA"

7
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:installation"
8
9
10

=== "AMD ROCm"

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

=== "Intel XPU"

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

## Requirements

- OS: Linux
20
- Python: 3.10 -- 3.13
21

22
23
24
!!! 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).

25
26
=== "NVIDIA CUDA"

27
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:requirements"
28
29
30

=== "AMD ROCm"

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

=== "Intel XPU"

35
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:requirements"
36
37
38
39
40
41
42
43
44

## Set up using Python

### Create a new Python environment

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

=== "NVIDIA CUDA"

45
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:set-up-using-python"
46
47
48

=== "AMD ROCm"

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

=== "Intel XPU"

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

### Pre-built wheels

=== "NVIDIA CUDA"

59
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:pre-built-wheels"
60
61
62

=== "AMD ROCm"

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

=== "Intel XPU"

67
    --8<-- "docs/getting_started/installation/gpu.xpu.inc.md:pre-built-wheels"
68
69
70
71
72
73
74

[](){ #build-from-source }

### Build wheel from source

=== "NVIDIA CUDA"

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

=== "AMD ROCm"

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

=== "Intel XPU"

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

## Set up using Docker

### Pre-built images

=== "NVIDIA CUDA"

91
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:pre-built-images"
92
93
94

=== "AMD ROCm"

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

=== "Intel XPU"

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

### Build image from source

=== "NVIDIA CUDA"

105
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:build-image-from-source"
106
107
108

=== "AMD ROCm"

109
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:build-image-from-source"
110
111
112

=== "Intel XPU"

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

## Supported features

=== "NVIDIA CUDA"

119
    --8<-- "docs/getting_started/installation/gpu.cuda.inc.md:supported-features"
120
121
122

=== "AMD ROCm"

123
    --8<-- "docs/getting_started/installation/gpu.rocm.inc.md:supported-features"
124
125
126

=== "Intel XPU"

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