opensource/DeOldify
3
0
Fork 0
mirror of https://github.com/jantic/DeOldify.git synced 2026-08-29 02:10:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #332 from macbre/patch-1

Browse Source 
Dockerfile pip install refactored
This commit is contained in:
Jason Antic
2021-04-16 10:37:16 -07:00
committed by GitHub
parent 9d72a9f12d c48a187fc9
commit 2b20617dea
1 changed files with 15 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+15 -19
View File
@@ -1,4 +1,4 @@
From nvcr.io/nvidia/pytorch:19.04-py3
FROM nvcr.io/nvidia/pytorch:19.04-py3
RUN apt-get -y update && apt-get install -y \
python3-pip \
@@ -20,24 +20,20 @@ RUN wget -O /root/.torch/models/resnet34-333f7ec4.pth https://download.pytorch.o
COPY Dockerfile ColorizeArtistic_gen.* /data/models/
COPY Dockerfile ColorizeVideo_gen.* /data/models/
RUN pip install --upgrade pip
RUN pip install versioneer==0.18
RUN pip install tensorboardX==1.6
RUN pip install Flask==1.1.1
RUN pip install pillow==6.1
RUN pip install numpy==1.16
RUN pip install scikit-image==0.15.0
RUN pip install requests==2.21.0
RUN pip install ffmpeg==1.4
RUN pip install ffmpeg-python==0.1.17
RUN pip install youtube-dl>=2019.4.17
RUN pip install jupyterlab==1.2.4
RUN pip install opencv-python>=3.3.0.10
RUN pip install fastai==1.0.51
RUN pip install --upgrade pip \
&& pip install versioneer==0.18 \
tensorboardX==1.6 \
Flask==1.1.1 \
pillow==6.1 \
numpy==1.16 \
scikit-image==0.15.0 \
requests==2.21.0 \
ffmpeg==1.4 \
ffmpeg-python==0.1.17 \
youtube-dl>=2019.4.17 \
jupyterlab==1.2.4 \
opencv-python>=3.3.0.10 \
fastai==1.0.51
ADD . /data/