unsloth/docker
Daniel Han 6a078b1a45 docker: close more pip-shim bypasses and make cu12.8 NVRTC the default
Notebook pip/uv shim (docker/unsloth_pip_shim.py, active only under
UNSLOTH_NB_SHIM=1):
  - Parse protected source archives (sdist/zip) by basename too, e.g.
    `pip install https://.../unsloth-2026.7.1.tar.gz` or `./torch-2.11.0.tar.gz`,
    mirroring the wheel-basename handling. A first-hyphen-before-digit split
    keeps hyphenated names like flashinfer-python intact.
  - Recognise uv's PLURAL long flags --requirements / --constraints, so those
    files go through the same protected-package filter as the singular names.
  - Drop --upgrade-strategy eager in shim mode so a kept target cannot eagerly
    rebuild already-satisfied baked deps (falls back to pip's only-if-needed).

NVRTC default (docker/Dockerfile, docker/Dockerfile.studio, docker/entrypoint.sh):
  - Make cu12.8 the immutable baked default (libnvrtc.so.12 -> .cu128.orig) with
    a staged .cu13 alias, and have select_cuda_jit_tools retarget to cu13 ONLY
    for sm_103/sm_121. Previously cu13 was baked as the default and restored to
    cu12.8 at runtime, so a non-root `docker run --user` container that cannot
    rewrite the symlink stayed on cu13 NVRTC and emitted cubins a 570-579 driver
    cannot load. The safe default now needs no runtime write.

Adds regression tests for each case (tests/python/test_unsloth_pip_shim.py,
tests/sh/test_select_cuda_jit_tools.sh).
2026-07-08 08:06:45 +00:00
..
jupyter docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
.dockerignore docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
build.sh docker: track latest llama.cpp release + show its update banner in Studio 2026-06-24 01:21:17 +00:00
docker_confirm.ps1 docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
docker_confirm.sh docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
Dockerfile docker: close more pip-shim bypasses and make cu12.8 NVRTC the default 2026-07-08 08:06:45 +00:00
Dockerfile.studio docker: close more pip-shim bypasses and make cu12.8 NVRTC the default 2026-07-08 08:06:45 +00:00
entrypoint.sh docker: close more pip-shim bypasses and make cu12.8 NVRTC the default 2026-07-08 08:06:45 +00:00
fetch_llama_prebuilt.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-06-26 05:42:43 +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
NOTICE docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
run.sh docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07: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: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
supervisord.conf docker: optional Cloudflare tunnel for JupyterLab (UNSLOTH_JUPYTER_CLOUDFLARE) 2026-06-24 09:55:11 +00:00
test_locally.sh docker: move the image torch stack to 2.11.0 and document the amd64 sm_103 JIT limit 2026-07-06 14:09:07 +00:00
unsloth_colab_compat.py docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
unsloth_ipython_startup.py docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
unsloth_jupyter_tunnel.sh docker: optional Cloudflare tunnel for JupyterLab (UNSLOTH_JUPYTER_CLOUDFLARE) 2026-06-24 09:55:11 +00:00
unsloth_llama_update.sh docker: add unsloth-llama-update for in-place llama.cpp prebuilt updates 2026-06-24 06:41:33 +00:00
unsloth_nb_compat.py docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
unsloth_nb_content_sig.py Docker notebook safety hardening and vLLM startup timeout fix 2026-07-05 13:53:39 +00:00
unsloth_nb_pip_magic.py [pre-commit.ci] auto fixes from pre-commit.com hooks 2026-06-27 08:47:23 +00:00
unsloth_nb_strip_colab.py docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
unsloth_nb_view.py docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00
unsloth_pip_shim.py docker: close more pip-shim bypasses and make cu12.8 NVRTC the default 2026-07-08 08:06:45 +00:00
unsloth_run.py Docker notebook safety hardening and vLLM startup timeout fix 2026-07-05 13:53:39 +00:00
unsloth_studio_update.sh docker: address review follow-ups (pip-shim flags, sync ownership, tags, zoo ref, arch list) 2026-06-26 09:06:39 +00:00
unsloth_sync_notebooks.sh docker: Colab-grade JupyterLab and Studio UX for the Unsloth image (#6681) 2026-07-06 08:27:00 -07:00