Pytorch-interactive/Dockerfile

9 lines
210 B
Docker
Raw Permalink Normal View History

2024-04-05 14:45:54 +00:00
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"]