NVLM-D-72B / Dockerfile
Uploading checkpoints, tokenizers, and license
0e2e85d
raw
history blame
241 Bytes
FROM nvcr.io/nvidia/pytorch:23.09-py3
RUN pip install transformers==4.39.3
RUN pip install accelerate==0.34.2
RUN pip install datasets==2.18.0
RUN pip install timm==1.0.9
RUN pip install anls==0.0.2
RUN pip install pycocoevalcap==1.2