From 59d33a26c7afc9f3ce38e4beefa99066d9c67ffa Mon Sep 17 00:00:00 2001
From: Lam Hieu <lamhieu.vk@gmail.com>
Date: Thu, 1 Feb 2024 16:50:37 +0700
Subject: [PATCH] Adding Cuda home environment values helps other libraries
 avoid unwanted errors, for example DeepSpeed

---
 Dockerfile.tmpl | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index 245228df..4e04b695 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -708,4 +708,6 @@ RUN echo "$GIT_COMMIT" > /etc/git_commit && echo "$BUILD_DATE" > /etc/build_date
 {{ if eq .Accelerator "gpu" }}
 # Remove the CUDA stubs.
 ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH_NO_STUBS"
+# Add the CUDA home.
+ENV CUDA_HOME=/usr/local/cuda
 {{ end }}