Add Dockerfile

iflytek · Sep 5, 2022 · 0a3e7e9 · 0a3e7e9
1 parent 50bf38c
commit 0a3e7e9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/iflearner/business/homo/Dockerfile b/iflearner/business/homo/Dockerfile
@@ -0,0 +1,10 @@
+FROM python:3.9
+
+ARG IFLEARNER_DIR=/iflearner
+COPY requirements.txt $IFLEARNER_DIR/
+COPY iflearner $IFLEARNER_DIR/iflearner
+COPY examples $IFLEARNER_DIR/examples
+ENV PYTHONPATH=$IFLEARNER_DIR
+WORKDIR $IFLEARNER_DIR
+RUN pip install -r requirements.txt -i https://pypi.douban.com/simple
+RUN pip install torch==1.8.1+cpu torchvision==0.9.1+cpu -f https://download.pytorch.org/whl/torch_stable.html