mirror of
https://github.com/NVIDIA/dgx-spark-playbooks.git
synced 2026-06-21 05:39:31 +00:00
17 lines
241 B
Docker
Executable File
17 lines
241 B
Docker
Executable File
FROM nvcr.io/nvidia/pytorch:26.04-py3
|
|
|
|
WORKDIR /workspace
|
|
|
|
RUN pip install \
|
|
datasets \
|
|
tokenizers \
|
|
wandb \
|
|
tiktoken \
|
|
psutil \
|
|
files-to-prompt \
|
|
uvicorn \
|
|
fastapi \
|
|
regex \
|
|
rustbpe
|
|
|
|
CMD ["/bin/bash"] |