From 259a8161f46a10dafe2a77b24fc2bacf57132f9b Mon Sep 17 00:00:00 2001 From: Tyler Perkins Date: Fri, 5 Apr 2024 10:45:54 -0400 Subject: [PATCH] Add complete dockerfile --- Dockerfile | 8 ++++++++ 1 file changed, 8 insertions(+) create mode 100644 Dockerfile diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..17bb889 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,8 @@ +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"]