[Bug] Python dependency issues for DGX Spark image

[sameersegal]

Hi,

I was excited to try out the DGX Spark FT tutorial on my new DGX Spark. However, it's not working because of python dependencies.

Once the image is created, import unsloth givens an error because torch gets replaced from the image's torch-2.9.0a0+50eac811a6.nv25.9 to torch-2.9.1-cp312-cp312-manylinux_2_28_aarch64.whl.

I tried to change the line in Dockerfile to --no-deps but this makes it difficult to find the right versions for all the dependencies:

RUN pip install --no-deps unsloth unsloth_zoo bitsandbytes==0.48.0 transformers==4.56.2 trl==0.22.2

What do you suggest I do?

Best,
Sameer