From 32a505d841df85e88e0183042c0cb24b688e7c8e Mon Sep 17 00:00:00 2001
From: sabaimran <narmiabas@gmail.com>
Date: Thu, 28 Mar 2024 11:37:37 +0530
Subject: [PATCH] Revert to using the nvidia base image for the next release

---
 prod.Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/prod.Dockerfile b/prod.Dockerfile
index 0da5363a..96aa9d3c 100644
--- a/prod.Dockerfile
+++ b/prod.Dockerfile
@@ -1,4 +1,5 @@
-FROM ubuntu:jammy
+FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
+# FROM ubuntu:jammy
 
 LABEL org.opencontainers.image.source https://github.com/khoj-ai/khoj