From 1e7cf09e889f5caa2d8b8c7165424cc56b449556 Mon Sep 17 00:00:00 2001 From: Darlan Nakamura Date: Mon, 26 Dec 2022 14:50:32 -0300 Subject: [PATCH] Fix Linux Docker container --- build/linux/Dockerfile | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/build/linux/Dockerfile b/build/linux/Dockerfile index b1cb6a6..8172b5f 100644 --- a/build/linux/Dockerfile +++ b/build/linux/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu20.04 +FROM nvidia/cuda:11.4.3-cudnn8-devel-ubuntu20.04 WORKDIR /app ARG DEBIAN_FRONTEND=noninteractive ARG NV_VER @@ -12,8 +12,6 @@ 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.4.0 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