fffiloni commited on
Commit
820e953
1 Parent(s): d3c9442

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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 -r 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
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