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

This commit is contained in:
milo157 2024-07-17 13:22:53 -04:00 committed by GitHub
parent 2fa4623d9e
commit a38524f338
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -28,6 +28,9 @@ Next, to install the required packages, add the following to your cerebrium.toml
.. code-block:: toml
[cerebrium.deployment]
docker_base_image_url = "nvidia/cuda:12.1.1-runtime-ubuntu22.04"
[cerebrium.dependencies.pip]
vllm = "latest"