From e2c13ff8666493f279484853bb7887c97bef6d9c Mon Sep 17 00:00:00 2001
From: tsubramanian <tamilselvan@gmail.com>
Date: Fri, 14 Jun 2024 16:28:35 -0400
Subject: [PATCH] Update Dockerfile

---
 devops/base/Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/devops/base/Dockerfile b/devops/base/Dockerfile
index 8101bc2be..a9e47e6b2 100644
--- a/devops/base/Dockerfile
+++ b/devops/base/Dockerfile
@@ -1,6 +1,6 @@
-ARG WBIA_UBUNTU_IMAGE=12.1.0-cudnn8-runtime-ubuntu22.04
+ARG WBIA_UBUNTU_IMAGE=nvidia/12.1.0-cudnn8-runtime-ubuntu22.04
 #nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
-docker pull nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
+#docker pull nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
 #nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
 #nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04