Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 # Use an official PyTorch image with CUDA support as the base image
-FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-runtime
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 # Install Git, OpenGL libraries, and libglib2.0
 RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0
@@ -35,7 +35,7 @@ WORKDIR $HOME/app
 RUN git clone -b main https://github.com/fffiloni/video-retalking $HOME/app
 
 # Install specific versions of PyTorch and TorchVision
-RUN pip install torch==1.9.0 torchvision==0.10.0
+RUN pip install torchvision==0.15.2
 
 # Install dependencies
 COPY requirements.txt $HOME/app/requirements.txt