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