Add Dockerfile. Modify script to build from the Dockerfile.

2023-05-24 14:02:45 +09:00
parent 869b4b8bc5
commit 51f2836df8
3 changed files with 24 additions and 36 deletions
@@ -0,0 +1,5 @@
+FROM python:3.11.3-slim-bullseye
+COPY requirements.txt /
+RUN apt update \
+    && apt install -y wget git \
+    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu117 --pre xformers