Skip to content

GitLab

Unverified Commit a38524f3 authored by milo157's avatar milo157 Committed by GitHub
Browse files

[DOC] - Add docker image to Cerebrium Integration (#6510)

parent 2fa4623d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
docs/source/serving/deploying_with_cerebrium.rst
View file @ a38524f3
...@@ -28,6 +28,9 @@ Next, to install the required packages, add the following to your cerebrium.toml ...@@ -28,6 +28,9 @@ Next, to install the required packages, add the following to your cerebrium.toml
.. code-block:: toml .. code-block:: toml
[cerebrium.deployment]
docker_base_image_url = "nvidia/cuda:12.1.1-runtime-ubuntu22.04"
[cerebrium.dependencies.pip] [cerebrium.dependencies.pip]
vllm = "latest" vllm = "latest"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment