diff --git a/Dockerfile.gpu b/Dockerfile.gpu index f940119f79..b18c58d447 100755 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -7,6 +7,7 @@ RUN add-apt-repository ppa:deadsnakes/ppa -y RUN apt-get update RUN apt install python3.8 -y RUN apt install python3.8-distutils -y +RUN apt install python3.8-tk -y RUN apt install curl -y RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py RUN python3.8 get-pip.py @@ -24,4 +25,4 @@ RUN jupyter serverextension enable --py jupyter_http_over_ws RUN alias python=python3.8 RUN echo "alias python=python3.8" >> /root/.bashrc WORKDIR "/notebooks" -CMD ["jupyter-notebook", "--allow-root" ,"--port=8888" ,"--no-browser" ,"--ip=0.0.0.0"] \ No newline at end of file +CMD ["jupyter-notebook", "--allow-root" ,"--port=8888" ,"--no-browser" ,"--ip=0.0.0.0"] diff --git a/INSTALL.md b/INSTALL.md index ef0836e05e..2a94d4f246 100755 --- a/INSTALL.md +++ b/INSTALL.md @@ -233,7 +233,9 @@ INFO 1. Install Docker deepfakes-gpu 1. Open a new terminal to interact with the project - docker exec faceswap-gpu python /srv/faceswap.py gui + docker exec -it deepfakes-gpu /bin/bash + # Launch deepfakes gui (Answer 3 for NVIDIA at the prompt) + python3.8 /srv/faceswap.py gui ``` A successful setup log, without docker.