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