Update Dockerfile
Browse files- Dockerfile +4 -1
Dockerfile
CHANGED
|
@@ -2,6 +2,9 @@ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
|
| 2 |
|
| 3 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 4 |
|
|
|
|
|
|
|
|
|
|
| 5 |
RUN apt-get update && apt-get install -y git wget ffmpeg
|
| 6 |
|
| 7 |
RUN useradd -m -u 1000 user
|
|
@@ -28,7 +31,7 @@ RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/
|
|
| 28 |
RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O $HOME/app/checkpoints/renderer.pt
|
| 29 |
|
| 30 |
# Install dependencies
|
| 31 |
-
RUN pip install --no-cache-dir
|
| 32 |
|
| 33 |
COPY app.py .
|
| 34 |
|
|
|
|
| 2 |
|
| 3 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 4 |
|
| 5 |
+
# Set the MKL_THREADING_LAYER environment variable to GNU
|
| 6 |
+
ENV MKL_THREADING_LAYER=GNU
|
| 7 |
+
|
| 8 |
RUN apt-get update && apt-get install -y git wget ffmpeg
|
| 9 |
|
| 10 |
RUN useradd -m -u 1000 user
|
|
|
|
| 31 |
RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O $HOME/app/checkpoints/renderer.pt
|
| 32 |
|
| 33 |
# Install dependencies
|
| 34 |
+
RUN pip install --no-cache-dir urllib3==1.26.6 transformers==4.28.1 dlib yacs scipy scikit-image scikit-learn PyYAML Pillow numpy opencv-python imageio ffmpeg-python av gradio
|
| 35 |
|
| 36 |
COPY app.py .
|
| 37 |
|