generated from tyler/Generic-Template
9 lines
210 B
Docker
9 lines
210 B
Docker
FROM python:3.10
|
|
|
|
RUN pip install torch torchvision torchaudio
|
|
RUN pip install transformers
|
|
RUN pip install numpy scipy
|
|
|
|
# Can also be set to /bin/sh if pip is needed first
|
|
ENTRYPOINT ["/usr/local/bin/python"]
|