Add CUDA_HOME for docker build

Author: liang2kl

docker/Dockerfile

@@ -44,6 +44,7 @@ COPY requirements.txt .
 
 RUN pip install -r requirements.txt --no-cache-dir
 
+ENV CUDA_HOME=/usr/local/cuda
 COPY ./kernels ./kernels
 RUN pip install ./kernels --no-cache-dir --no-build-isolation
 

docker/Dockerfile.chat

@@ -41,6 +41,7 @@ ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0 10.0 12.0"
 
 RUN pip install vllm==0.15.1 accelerate --no-cache-dir
 
+ENV CUDA_HOME=/usr/local/cuda
 COPY ./kernels ./kernels
 RUN pip install ./kernels --no-cache-dir --no-build-isolation
 

docker/Dockerfile.chat-cu130

@@ -41,6 +41,7 @@ ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0 10.0 12.0 12.1"
 
 RUN pip install vllm==0.15.1 accelerate --no-cache-dir --extra-index-url https://wheels.vllm.ai/0.15.1/cu130 --extra-index-url https://download.pytorch.org/whl/cu130
 
+ENV CUDA_HOME=/usr/local/cuda
 COPY ./kernels ./kernels
 RUN pip install ./kernels --no-cache-dir --no-build-isolation
 

docker/Dockerfile.eval-reasoning

@@ -43,6 +43,7 @@ COPY experiments/tasks/reasoning/requirements.txt .
 
 RUN pip install -r ./requirements.txt --no-cache-dir
 
+ENV CUDA_HOME=/usr/local/cuda
 COPY ./kernels ./kernels
 RUN pip install ./kernels --no-cache-dir --no-build-isolation