mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-06-24 07:09:31 +00:00
11 lines
423 B
Docker
11 lines
423 B
Docker
|
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"]
|