From 32dd412b608246db074559df0d23e35875085947 Mon Sep 17 00:00:00 2001 From: tsubramanian Date: Sat, 15 Jun 2024 15:56:43 -0400 Subject: [PATCH] Update Dockerfile --- devops/base/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/devops/base/Dockerfile b/devops/base/Dockerfile index 491ccf46a..9b470cbb7 100644 --- a/devops/base/Dockerfile +++ b/devops/base/Dockerfile @@ -1,4 +1,4 @@ -ARG WBIA_UBUNTU_IMAGE=nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04 +ARG WBIA_UBUNTU_IMAGE=nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 #nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 #nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04 #nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04