fffiloni commited on
Commit
233848d
1 Parent(s): f21f318

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,5 +1,5 @@
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 OpenGL libraries, and libglib2.0
5
  RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0
@@ -31,7 +31,7 @@ WORKDIR $HOME/app
31
  RUN git clone -b main https://github.com/fffiloni/video-retalking $HOME/app
32
 
33
  # Install specific versions of PyTorch, TorchVision
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
 
1
  # Use an official PyTorch image with CUDA support as the base image
2
+ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
3
 
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
 
31
  RUN git clone -b main https://github.com/fffiloni/video-retalking $HOME/app
32
 
33
  # Install specific versions of PyTorch, TorchVision
34
+ RUN pip install torch==2.0.1 torchvision==0.15.2
35
 
36
  # Install dependencies
37
  RUN pip install --no-cache-dir -r requirements.txt