Update llama-server-cuda.Dockerfile
This commit is contained in:
parent
253b7fde91
commit
d7da7ff8de
1 changed files with 2 additions and 2 deletions
|
@ -1,6 +1,6 @@
|
||||||
ARG UBUNTU_VERSION=22.04
|
ARG UBUNTU_VERSION=20.04
|
||||||
# This needs to generally match the container host's environment.
|
# This needs to generally match the container host's environment.
|
||||||
ARG CUDA_VERSION=12.6.0
|
ARG CUDA_VERSION=11.3.1
|
||||||
# Target the CUDA build image
|
# Target the CUDA build image
|
||||||
ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
|
ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
|
||||||
# Target the CUDA runtime image
|
# Target the CUDA runtime image
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue