From 9cbf46847f990fb43a289fd2782337032e4cba2b Mon Sep 17 00:00:00 2001
From: luv-bansal <bansal22luvi@gmail.com>
Date: Thu, 10 Oct 2024 08:29:20 -0400
Subject: [PATCH] updated cuda dockerfile

---
 clarifai/runners/dockerfile_template/Dockerfile.cuda.template | 1 +
 1 file changed, 1 insertion(+)

diff --git a/clarifai/runners/dockerfile_template/Dockerfile.cuda.template b/clarifai/runners/dockerfile_template/Dockerfile.cuda.template
index 9aae96d..4ac2243 100644
--- a/clarifai/runners/dockerfile_template/Dockerfile.cuda.template
+++ b/clarifai/runners/dockerfile_template/Dockerfile.cuda.template
@@ -66,6 +66,7 @@ ENV CLARIFAI_API_BASE=${CLARIFAI_API_BASE}
 
 # Set the NUMBA cache dir to /tmp
 ENV NUMBA_CACHE_DIR=/tmp/numba_cache
+ENV HOME=/tmp
 
 # Set the working directory to /app
 WORKDIR /app