Sébastien De Greef commited on
Commit
0561241
·
1 Parent(s): 7fa050d

chore: Update Dockerfile to install cuDNN library and fix installation command

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -29,10 +29,10 @@ RUN rm -f /etc/apt/sources.list.d/*.list && \
29
  && rm -rf /var/lib/apt/lists/*
30
 
31
 
32
- RUN wget https://developer.download.nvidia.com/compute/cudnn/9.1.1/local_installers/cudnn-local-repo-ubuntu2204-9.1.1_1.0-1_amd64.deb \
33
  && sudo dpkg -i cudnn-local-repo-ubuntu2204-9.1.1_1.0-1_amd64.deb \
34
  && sudo cp /var/cudnn-local-repo-ubuntu2204-9.1.1/cudnn-*-keyring.gpg /usr/share/keyrings/ \
35
- && sudo apt-get update && sudo apt-get -y install cuda cudnn
36
 
37
  # Create a working directory
38
  WORKDIR /app
 
29
  && rm -rf /var/lib/apt/lists/*
30
 
31
 
32
+ RUN wget -q https://developer.download.nvidia.com/compute/cudnn/9.1.1/local_installers/cudnn-local-repo-ubuntu2204-9.1.1_1.0-1_amd64.deb \
33
  && sudo dpkg -i cudnn-local-repo-ubuntu2204-9.1.1_1.0-1_amd64.deb \
34
  && sudo cp /var/cudnn-local-repo-ubuntu2204-9.1.1/cudnn-*-keyring.gpg /usr/share/keyrings/ \
35
+ && sudo apt-get update && sudo apt-get -y install cudnn
36
 
37
  # Create a working directory
38
  WORKDIR /app