updated cuda dockerfile

Clarifai · Oct 10, 2024 · 9cbf468 · 9cbf468
1 parent 326e835
commit 9cbf468
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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