Gregniuki commited on
Commit
bd26fb6
·
1 Parent(s): e4345d4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -64,7 +64,8 @@ RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/
64
 
65
  # Install FFmpeg
66
  RUN apt-get update && \
67
- apt-get install -y python3-pip ffmpeg libcublas-11-8
 
68
 
69
 
70
  # Use the official Python 3.9 image as the base image
 
64
 
65
  # Install FFmpeg
66
  RUN apt-get update && \
67
+ apt-get install -y python3-pip ffmpeg libcublas-11-8 libcudnn8
68
+
69
 
70
 
71
  # Use the official Python 3.9 image as the base image