mirror of
https://github.com/deepfakes/faceswap.git
synced 2026-10-23 10:22:29 +08:00
Alessandro Melis
23 lines
795 B
Docker
Executable File
23 lines
795 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 git \
|
|
&& 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"]
|