8000 Set CUDA_VERSION as build ARG · HardSoft2023/llama-cpp-python@6ece8a2 · GitHub
[go: up one dir, main page]
Skip to content

Commit 6ece8a2

Browse files
PipboyguyPipboyguy
committed
Set CUDA_VERSION as build ARG
Signed-off-by: Marcel Coetzee <marcel@mooncoon.com>
1 parent 6c57d38 commit 6ece8a2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

‎Dockerfile.cuda

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
FROM nvidia/cuda:12.0.1-devel-ubuntu22.04
1+
ARG CUDA_VERSION=12.1.1
2+
FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
23

34
# We need to set the host to 0.0.0.0 to allow outside access
45
ENV HOST 0.0.0.0

0 commit comments

Comments
 (0)
0