fffiloni commited on
Commit
cb8e31c
1 Parent(s): eb8825f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -4,8 +4,8 @@ FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
4
  # Install Git and OpenGL libraries, and libglib2.0
5
  RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0
6
 
7
- # Install the unzip and ffmpeg utilities (if not already installed)
8
- RUN apt-get update && apt-get install -y unzip ffmpeg
9
 
10
  # Set up a new user named "user" with user ID 1000
11
  RUN useradd -m -u 1000 user
@@ -34,7 +34,7 @@ RUN git clone -b main https://github.com/fffiloni/video-retalking $HOME/app
34
  RUN pip install torch==1.9.0 torchvision==0.10.0
35
 
36
  # Install dependencies
37
- RUN pip install --no-cache-dir -r requirements.txt gradio
38
 
39
  USER root
40
  # Update package lists and install aria2 without sudo
 
4
  # Install Git and OpenGL libraries, and libglib2.0
5
  RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0
6
 
7
+ # Install necessary dependencies, including CMake
8
+ RUN apt-get update && apt-get install -y unzip ffmpeg cmake
9
 
10
  # Set up a new user named "user" with user ID 1000
11
  RUN useradd -m -u 1000 user
 
34
  RUN pip install torch==1.9.0 torchvision==0.10.0
35
 
36
  # Install dependencies
37
+ RUN pip install --no-cache-dir -r requirements.txt
38
 
39
  USER root
40
  # Update package lists and install aria2 without sudo