github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-20 13:33:25 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #91 from CeeBeeEh/master

Browse source 
OpenCL fix for Linux Docker
This commit is contained in:
iperov 2022-10-01 09:36:40 +04:00 • committed by GitHub
commit 32202d68c2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
build/linux/Dockerfile
View file

@ -1,6 +1,7 @@
FROM nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu20.04 FROM nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu20.04
WORKDIR /app WORKDIR /app
ARG DEBIAN_FRONTEND=noninteractive ARG DEBIAN_FRONTEND=noninteractive
ARG NV_VER
RUN apt update 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 -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 apt install -y python3 python3-pip
@ -11,6 +12,8 @@ RUN git clone https://github.com/iperov/DeepFaceLive.git
RUN python -m pip install --upgrade pip RUN python -m pip install --upgrade pip
RUN python -m pip install onnxruntime-gpu==1.12.1 numpy==1.21.6 h5py numexpr protobuf==3.20.1 opencv-python==4.6.0.66 opencv-contrib-python==4.6.0.66 pyqt6==6.3.1 onnx==1.12.0 torch==1.10.0 torchvision==0.11.1 RUN python -m pip install onnxruntime-gpu==1.12.1 numpy==1.21.6 h5py numexpr protobuf==3.20.1 opencv-python==4.6.0.66 opencv-contrib-python==4.6.0.66 pyqt6==6.3.1 onnx==1.12.0 torch==1.10.0 torchvision==0.11.1
RUN apt install -y libnvidia-compute-$NV_VER
WORKDIR /app/DeepFaceLive WORKDIR /app/DeepFaceLive
COPY example.sh example.sh COPY example.sh example.sh
CMD ./example.sh CMD ./example.sh

5
build/linux/start.sh
View file

@ -1,5 +1,8 @@
#!/bin/bash #!/bin/bash
NV_LIB=$(locate nvidia.ko |grep $(uname -r) |grep dkms | head -1)
NV_VER=$(modinfo $NV_LIB | grep ^version |awk '{print $2}'|awk -F '.' '{print $1}')
DATA_FOLDER=$(pwd)/data/ DATA_FOLDER=$(pwd)/data/
declare CAM0 CAM1 CAM2 CAM3 declare CAM0 CAM1 CAM2 CAM3
printf "\n" printf "\n"
@ -28,6 +31,6 @@ shift "$(($OPTIND -1))"
printf "\n" printf "\n"
# Warning xhost + is overly permissive and will reduce system security. Edit as desired # Warning xhost + is overly permissive and will reduce system security. Edit as desired
docker build . -t deepfacelive docker build . -t deepfacelive --build-arg NV_VER=$NV_VER
xhost + xhost +
docker run --ipc host --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix -v $DATA_FOLDER:/data/ $CAM0 $CAM1 $CAM2 $CAM3 --rm -it deepfacelive docker run --ipc host --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix -v $DATA_FOLDER:/data/ $CAM0 $CAM1 $CAM2 $CAM3 --rm -it deepfacelive