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"]
|