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
### Pre-built wheels {#pre-built-wheels}
56
57
58

=== "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

### Build wheel from source

=== "NVIDIA CUDA"

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

=== "AMD ROCm"

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

=== "Intel XPU"

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

## Set up using Docker

### Pre-built images

=== "NVIDIA CUDA"

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

=== "AMD ROCm"

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

=== "Intel XPU"

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

### Build image from source

=== "NVIDIA CUDA"

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

=== "AMD ROCm"

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

=== "Intel XPU"

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

## Supported features

=== "NVIDIA CUDA"

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

=== "AMD ROCm"

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

=== "Intel XPU"

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