unsloth/docker
danielhanchen e8ac40fa5b docker/studio: deterministic Studio install inside the image build
Two failures from the first in-image Studio install, both rooted in
install.sh probing the build host:

1. setup.sh aborted on the pre-linked llama.cpp dir: 'already exists and
   is not marked as a Studio-owned llama.cpp install'. The dir is the
   image's baked prebuilt, provisioned exclusively for Studio, so write
   the .unsloth-studio-owned marker next to the binaries.

2. With no GPU and no nvidia-smi in the build container, install.sh fell
   back to cu126 torch wheels for the Studio venv (and would pick cpu
   wheels on a CI runner without /proc/driver/nvidia), so the published
   image's Studio venv would depend on which host built it and could not
   train on Blackwell. get_torch_index_url now honours an explicit
   UNSLOTH_TORCH_INDEX_FAMILY override naming the index leaf (cu128,
   cu130, rocm7.2, cpu, ...). The resolved family flows into
   UNSLOTH_TORCH_BACKEND, which install_python_stack.py already consumes,
   so the whole downstream chain follows the pin. Dockerfile.studio sets
   cu128 on amd64 and cu130 on arm64 (DGX Spark / Grace).
2026-06-12 05:06:51 +00:00
..
.dockerignore docker: whitelist new build-context files, add docker_confirm.ps1 2026-06-12 05:06:51 +00:00
build.sh Add Docker build for Blackwell that runs on any NVIDIA GPU host 2026-05-24 06:52:58 +00:00
docker_confirm.ps1 docker: whitelist new build-context files, add docker_confirm.ps1 2026-06-12 05:06:51 +00:00
docker_confirm.sh docker: pin the llama.cpp bake by target arch, add docker_confirm.sh 2026-06-12 05:06:51 +00:00
Dockerfile docker: pin the llama.cpp bake by target arch, add docker_confirm.sh 2026-06-12 05:06:51 +00:00
Dockerfile.studio docker/studio: deterministic Studio install inside the image build 2026-06-12 05:06:51 +00:00
entrypoint.sh docker: ship Jupyter, Studio and prebuilt llama.cpp out of the box 2026-06-12 05:06:51 +00:00
fetch_llama_prebuilt.py docker: pin the llama.cpp bake by target arch, add docker_confirm.sh 2026-06-12 05:06:51 +00:00
freeze.sh Add Docker build for Blackwell that runs on any NVIDIA GPU host 2026-05-24 06:52:58 +00:00
hf_pull.sh hf_{push,pull}.sh: use new hf CLI, fall back to deprecated huggingface-cli 2026-05-24 09:40:22 +00:00
hf_push.sh hf_{push,pull}.sh: use new hf CLI, fall back to deprecated huggingface-cli 2026-05-24 09:40:22 +00:00
run.sh docker: ship Jupyter, Studio and prebuilt llama.cpp out of the box 2026-06-12 05:06:51 +00:00
setup_qemu.sh Add linux/arm64 (DGX Spark / Grace) support via QEMU at build time 2026-05-24 10:34:59 +00:00
smoke_test.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-06-12 04:35:41 +00:00
studio_launch.sh docker: ship Jupyter, Studio and prebuilt llama.cpp out of the box 2026-06-12 05:06:51 +00:00
supervisord.conf docker: ship Jupyter, Studio and prebuilt llama.cpp out of the box 2026-06-12 05:06:51 +00:00
test_locally.sh docker/test_locally.sh: fail fast + pin notebook fetch to immutable SHA 2026-05-27 15:56:30 +00:00