Files
faceswap/Dockerfile.gpu
T
Timothy Delter a90eff901d Fix for Effmpeg not working when using Faceswap with Docker (#732)
* Edited Dockerfiles to include ffmpeg installation for Effmpeg to work

* Edited incorrect command to run Dockerized Faceswap GUI in INSTALL.md
2019-05-18 11:21:02 +01:00

23 lines
791 B
Docker
Executable File

FROM tensorflow/tensorflow:1.12.0-gpu-py3
RUN add-apt-repository -y ppa:jonathonf/ffmpeg-4 \
&& apt-get update -qq -y \
&& apt-get install -y libsm6 libxrender1 libxext-dev python3-tk ffmpeg \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*
COPY requirements.txt /opt/
RUN pip3 install --upgrade pip
RUN pip3 --no-cache-dir install -r /opt/requirements.txt && rm /opt/requirements.txt
RUN pip3 install jupyter matplotlib
RUN pip3 install jupyter_http_over_ws
RUN jupyter serverextension enable --py jupyter_http_over_ws
# patch for tensorflow:latest-gpu-py3 image
RUN cd /usr/local/cuda/lib64 \
&& mv stubs/libcuda.so ./ \
&& ln -s libcuda.so libcuda.so.1 \
&& ldconfig
WORKDIR "/notebooks"
CMD ["jupyter-notebook", "--allow-root" ,"--port=8888" ,"--no-browser" ,"--ip=0.0.0.0"]