File size: 241 Bytes
b5731cf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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
|