Hugging Face's logo

Spaces:
fffiloni
/
VideoRetalking
Running on T4

App Files Community
5
fffiloni commited on
Commit
27772e0
·
1 Parent(s): c1436f4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
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 system libraries required for OpenGL without interactive prompts
5
  ENV DEBIAN_FRONTEND=noninteractive
 
1
  # Use an official PyTorch image with CUDA support as the base image
2
+ FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-devel
3
 
4
  # Install Git and system libraries required for OpenGL without interactive prompts
5
  ENV DEBIAN_FRONTEND=noninteractive