mirror of
https://github.com/iperov/DeepFaceLive
synced 2025-08-14 10:47:00 -07:00
Updated Dockerfile and start.sh to enable OpenCL
This commit is contained in:
parent
6f28fdd15d
commit
05c9faef36
1 changed files with 3 additions and 0 deletions
|
@ -1,6 +1,7 @@
|
|||
FROM nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu20.04
|
||||
WORKDIR /app
|
||||
ARG DEBIAN_FRONTEND=noninteractive
|
||||
ARG NV_VER
|
||||
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
|
||||
|
@ -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 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
|
||||
COPY example.sh example.sh
|
||||
CMD ./example.sh
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue