tyler
/
Pytorch-interactive
generated from tyler/Generic-Template
1
1
Fork 0
Code Issues Pull Requests Packages Releases Activity

Add complete dockerfile

This commit is contained in:
Tyler Perkins 2024-04-05 10:45:54 -04:00
parent ae85b28650
commit 259a8161f4
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Dockerfile Normal file
View File

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