Merge pull request #120 from tariq1890/bump-cuda-base

bump cuda base image to 12.6.2-ubi9
NVIDIA · Oct 15, 2024 · bd30d3f · bd30d3f
2 parents e8ab1e3 + ba8a19d
commit bd30d3f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -25,7 +25,7 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 ARG CUDA_IMAGE=cuda
-ARG CUDA_VERSION=12.6.1
+ARG CUDA_VERSION=12.6.2
 ARG BASE_DIST=ubi9
 
 FROM nvcr.io/nvidia/${CUDA_IMAGE}:${CUDA_VERSION}-base-${BASE_DIST} as builder
@@ -48,7 +48,7 @@ COPY . .
 
 RUN make build
 
-FROM nvcr.io/nvidia/cuda:12.6.1-base-ubi9
+FROM nvcr.io/nvidia/${CUDA_IMAGE}:${CUDA_VERSION}-base-${BASE_DIST}
 
 ARG VERSION