Sébastien De Greef commited on
Commit
e469db9
·
1 Parent(s): 85bf4d5

chore: Update Dockerfile to install CUDA 12.4.1 and cuDNN 9.1.1 libraries

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -4
Dockerfile CHANGED
@@ -2,10 +2,11 @@ FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
  ENV TZ=Europe/Paris
 
5
 
6
  # Install some basic utilities
7
  RUN rm -f /etc/apt/sources.list.d/*.list && \
8
- apt-get update && apt-get install -y \
9
  curl \
10
  ca-certificates \
11
  sudo \
@@ -30,10 +31,10 @@ RUN rm -f /etc/apt/sources.list.d/*.list && \
30
 
31
  ENV PATH="/home/chatbot/.local/bin:${PATH}"
32
 
33
- RUN wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run
34
- RUN sudo sh cuda_12.4.1_550.54.15_linux.run
35
 
36
- 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 \
37
  && sudo dpkg -i cudnn-local-repo-ubuntu2204-9.1.1_1.0-1_amd64.deb \
38
  && sudo cp /var/cudnn-local-repo-ubuntu2204-9.1.1/cudnn-*-keyring.gpg /usr/share/keyrings/ \
39
  && sudo apt-get update && sudo apt-get -y install cudnn
 
2
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
  ENV TZ=Europe/Paris
5
+ ENV TERM=linux
6
 
7
  # Install some basic utilities
8
  RUN rm -f /etc/apt/sources.list.d/*.list && \
9
+ apt-get update && apt-get install -y -qq \
10
  curl \
11
  ca-certificates \
12
  sudo \
 
31
 
32
  ENV PATH="/home/chatbot/.local/bin:${PATH}"
33
 
34
+ RUN wget -q https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run
35
+ RUN sudo bash cuda_12.4.1_550.54.15_linux.run
36
 
37
+ 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 \
38
  && sudo dpkg -i cudnn-local-repo-ubuntu2204-9.1.1_1.0-1_amd64.deb \
39
  && sudo cp /var/cudnn-local-repo-ubuntu2204-9.1.1/cudnn-*-keyring.gpg /usr/share/keyrings/ \
40
  && sudo apt-get update && sudo apt-get -y install cudnn