Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Use an official PyTorch image with CUDA support as the base image
|
| 2 |
-
FROM pytorch/pytorch:
|
| 3 |
|
| 4 |
# Install Git and system libraries required for OpenGL without interactive prompts
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive
|
|
@@ -40,11 +40,11 @@ WORKDIR $HOME/app
|
|
| 40 |
RUN git clone -b main https://github.com/fffiloni/video-retalking $HOME/app
|
| 41 |
|
| 42 |
# Install specific versions of PyTorch and TorchVision
|
| 43 |
-
RUN pip install torchvision==0.
|
| 44 |
|
| 45 |
# Install dependencies
|
| 46 |
#COPY requirements.txt $HOME/app/requirements.txt
|
| 47 |
-
RUN pip install --no-cache-dir -r requirements.txt gradio opencv-python scipy
|
| 48 |
|
| 49 |
# Download checkpoint files using aria2
|
| 50 |
RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/video-retalking/resolve/main/30_net_gen.pth -d $HOME/app/checkpoints -o 30_net_gen.pth
|
|
|
|
| 1 |
# Use an official PyTorch image with CUDA support as the base image
|
| 2 |
+
FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
|
| 3 |
|
| 4 |
# Install Git and system libraries required for OpenGL without interactive prompts
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
| 40 |
RUN git clone -b main https://github.com/fffiloni/video-retalking $HOME/app
|
| 41 |
|
| 42 |
# Install specific versions of PyTorch and TorchVision
|
| 43 |
+
RUN pip install torchvision==0.10.0
|
| 44 |
|
| 45 |
# Install dependencies
|
| 46 |
#COPY requirements.txt $HOME/app/requirements.txt
|
| 47 |
+
RUN pip install --no-cache-dir -r requirements.txt gradio==3.7.0 opencv-python scipy
|
| 48 |
|
| 49 |
# Download checkpoint files using aria2
|
| 50 |
RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/video-retalking/resolve/main/30_net_gen.pth -d $HOME/app/checkpoints -o 30_net_gen.pth
|