From bd31f69a2cf05f0e5be4e3d63551617bae5561e9 Mon Sep 17 00:00:00 2001
From: Sameer Segal <sameersegal@gmail.com>
Date: Wed, 19 Nov 2025 21:24:15 +0530
Subject: [PATCH] Update CUDA_HOME and install dependencies

These tweaks seem to work with DGX Spark
---
 Dockerfile_DGX_Spark | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/Dockerfile_DGX_Spark b/Dockerfile_DGX_Spark
index 45096b5d..be79911f 100644
--- a/Dockerfile_DGX_Spark
+++ b/Dockerfile_DGX_Spark
@@ -1,7 +1,7 @@
 FROM nvcr.io/nvidia/pytorch:25.09-py3
 
 # Set CUDA environment variables
-ENV CUDA_HOME=/usr/local/cuda-13.0/
+ENV CUDA_HOME=/usr/local/cuda-13.0
 ENV CUDA_PATH=$CUDA_HOME
 ENV PATH=$CUDA_HOME/bin:$PATH
 ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
@@ -24,7 +24,8 @@ RUN git clone --depth=1 https://github.com/facebookresearch/xformers --recursive
     cd ..
 
 # Install unsloth and other dependencies
-RUN pip install unsloth unsloth_zoo bitsandbytes==0.48.0 transformers==4.56.2 trl==0.22.2
+RUN pip install --no-deps bitsandbytes==0.48.0 transformers==4.56.2 trl==0.22.2
+RUN pip install unsloth unsloth_zoo
 
 # Launch the shell
-CMD ["/bin/bash"]
\ No newline at end of file
+CMD ["/bin/bash"]