-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
pookie
authored and
pookie
committed
Sep 22, 2021
1 parent
a327638
commit 4b30919
Showing
4 changed files
with
38 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
FROM nvidia/cuda:11.4.2-cudnn8-devel-ubuntu20.04 | ||
WORKDIR /app | ||
ARG DEBIAN_FRONTEND=noninteractive | ||
RUN apt update | ||
RUN apt -y install libgl1-mesa-glx libegl1-mesa libxrandr2 libxrandr2 libxss1 libxcursor1 libxcomposite1 libasound2 libxi6 libxtst6 curl ffmpeg git nano gnupg2 libsm6 wget unzip libxcb-icccm4 libxkbcommon-x11-0 libxcb-keysyms1 libxcb-icccm4 libxcb-render0 libxcb-render-util0 libxcb-image0 | ||
RUN apt install -y python3 python3-pip | ||
RUN ln -s /usr/bin/python3 /usr/bin/python | ||
|
||
RUN git clone https://github.com/iperov/DeepFaceLive.git | ||
|
||
RUN python -m pip install --upgrade pip | ||
RUN python -m pip install onnxruntime-gpu==1.8.1 cupy-cuda113 numpy==1.21.2 scipy==1.5.4 numexpr opencv-python==4.5.3.56 opencv-contrib-python==4.5.3.56 pyqt6==6.1.1 onnx==1.10.1 torch==1.8.1 torchvision==0.9.1 | ||
|
||
WORKDIR /app/DeepFaceLive | ||
COPY example.sh example.sh | ||
CMD ./example.sh |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,15 @@ | ||
# DeepFaceLive Docker | ||
|
||
### Requirements | ||
|
||
* nvidia-driver-470 (current) | ||
* cuda 11.4 (current) | ||
|
||
### Setup | ||
|
||
``` | ||
git clone DeepFaceLive | ||
cd DeepFaceLive/build/linux/ | ||
# start.sh builds and runs container with proper options, then example.sh will start DeepFaceLive with example options | ||
./start.sh | ||
``` |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,2 @@ | ||
#!/bin/bash | ||
python main.py run DeepFaceLive --userdata-dir /data/ |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
#!/bin/bash | ||
# Warning xhost + is overly permissive and will reduce system security. Edit as desired | ||
docker build . -t deepfacelive | ||
xhost + | ||
docker run --ipc host --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix -v $(pwd)/data/:/data/ --rm -it deepfacelive |