From 6ece8a225a9cad89eec41f08744e3730c600ebcd Mon Sep 17 00:00:00 2001 From: Marcel Coetzee Date: Thu, 18 May 2023 16:59:42 +0200 Subject: [PATCH] Set CUDA_VERSION as build ARG Signed-off-by: Marcel Coetzee --- Dockerfile.cuda | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Dockerfile.cuda b/Dockerfile.cuda index d9451bc..e95fa23 100644 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -1,4 +1,5 @@ -FROM nvidia/cuda:12.0.1-devel-ubuntu22.04 +ARG CUDA_VERSION=12.1.1 +FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 # We need to set the host to 0.0.0.0 to allow outside access ENV HOST 0.0.0.0