From f9e8d343eed38bb1a6ab0500d7bf1560d825fc79 Mon Sep 17 00:00:00 2001
From: Joseph Young <130690493+yousecjoe@users.noreply.github.com>
Date: Fri, 15 Mar 2024 04:32:10 -0400
Subject: [PATCH] Switch base image to nvidia/cuda for GPU support

Updated Dockerfile to use nvidia/cuda:12.3.2-base-ubuntu22.04 as the base image to enable GPU support. This change is made for better compatibility with CUDA 12 and GPU acceleration, aligning with the requirement to run applications that make extensive use of GPU resources. The previous image from chainguard was replaced to exploit the latest CUDA version's features and performance improvements. The comment on the ubuntu:22.04 line preserves the option for a future switch if needed.

Refs #1234 (if there's an associated issue)
---
 Dockerfile | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index fba49a9c07..6cc54b9939 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -16,7 +16,9 @@ RUN npm run build
 
 
 #FROM python:3.11-slim-bookworm as base
-FROM cgr.dev/chainguard/pytorch-cuda12 as base
+#FROM cgr.dev/chainguard/pytorch-cuda12 as base
+#FROM ubuntu:22.04 as base
+FROM nvidia/cuda:12.3.2-base-ubuntu22.04 as base
 
 ENV ENV=prod
 ENV PORT ""