mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-06-21 21:59:30 +00:00
11 lines
423 B
Docker
Executable File
11 lines
423 B
Docker
Executable File
FROM nvcr.io/nvidia/pytorch:25.09-py3
|
|
|
|
WORKDIR /workspace
|
|
|
|
# Install dependencies globally so torchrun (which uses /usr/bin/python) can access them
|
|
RUN /usr/bin/python -m pip install tiktoken tokenizers datasets psutil files-to-prompt regex setuptools uvicorn wandb maturin
|
|
|
|
# Create venv with --system-site-packages so it inherits global packages
|
|
RUN /usr/bin/python -m venv --system-site-packages .venv
|
|
|
|
CMD ["/bin/bash"] |