Skip to content

Commit 44f3a70

Browse files
milo157jimpang
authored andcommitted
[DOC] - Add docker image to Cerebrium Integration (vllm-project#6510)
1 parent f00b84f commit 44f3a70

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

docs/source/serving/deploying_with_cerebrium.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,9 @@ Next, to install the required packages, add the following to your cerebrium.toml
2828

2929
.. code-block:: toml
3030
31+
[cerebrium.deployment]
32+
docker_base_image_url = "nvidia/cuda:12.1.1-runtime-ubuntu22.04"
33+
3134
[cerebrium.dependencies.pip]
3235
vllm = "latest"
3336

0 commit comments

Comments
 (0)