Skip to content

Commit 01c79e7

Browse files
abetlenabetlen
authored
Merge pull request abetlen#258 from Pipboyguy/main
Change docker build dynamic param to image instead of cuda version
2 parents c3e80b1 + e6639e6 commit 01c79e7

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎Dockerfile.cuda‎

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

44
# We need to set the host to 0.0.0.0 to allow outside access
55
ENV HOST 0.0.0.0

0 commit comments

Comments
 (0)