Gregniuki commited on
Commit
d405a2b
·
1 Parent(s): dc35509

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -63,6 +63,7 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machi
63
  RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" > /etc/apt/sources.list.d/cuda.list
64
 
65
  # Install FFmpeg
 
66
  RUN apt-get update && \
67
  apt-get install -y python3-pip ffmpeg libcublas-11-8 libcudnn8=8.7.0.84+cuda11.8
68
 
 
63
  RUN echo "deb https://developer.download.nvidia.com/compute/cuda/repos/debian11/x86_64 /" > /etc/apt/sources.list.d/cuda.list
64
 
65
  # Install FFmpeg
66
+ RUN apt list -a libcudnn8-dev
67
  RUN apt-get update && \
68
  apt-get install -y python3-pip ffmpeg libcublas-11-8 libcudnn8=8.7.0.84+cuda11.8
69