unsloth/studio/backend/core/inference
Daniel Han f1293fe7d8 studio: respect existing CUDA_VISIBLE_DEVICES in GPU selection
If CUDA_VISIBLE_DEVICES is already set in the environment (e.g.,
by the user or a wrapper script), only consider those GPUs when
selecting devices for llama-server. nvidia-smi reports all physical
GPUs regardless of CUDA_VISIBLE_DEVICES, so we filter its output
to match the allowed set.

Without this, the GPU selector could pick a GPU outside the user's
allowed set, overriding their restriction.
2026-03-15 05:24:06 -07:00
..
__init__.py Final cleanup 2026-03-12 18:28:04 +00:00
audio_codecs.py Final cleanup 2026-03-12 18:28:04 +00:00
inference.py Fix: Compare Mode Deadlock, Cancel Event Poisoning & IPC Optimization (#4303) 2026-03-15 16:11:44 +04:00
llama_cpp.py studio: respect existing CUDA_VISIBLE_DEVICES in GPU selection 2026-03-15 05:24:06 -07:00
orchestrator.py Fix: Compare Mode Deadlock, Cancel Event Poisoning & IPC Optimization (#4303) 2026-03-15 16:11:44 +04:00
worker.py Final cleanup 2026-03-12 18:28:04 +00:00