fffiloni commited on
Commit
013a047
·
verified ·
1 Parent(s): 692d4e7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -2
Dockerfile CHANGED
@@ -27,7 +27,7 @@ WORKDIR $HOME/app
27
 
28
  # Install system dependencies as root
29
  USER root
30
- # Install basic utilities and Python 3.9
31
  RUN apt-get update && apt-get install -y \
32
  wget \
33
  curl \
@@ -43,7 +43,14 @@ RUN apt-get update && apt-get install -y \
43
  && curl -sS https://bootstrap.pypa.io/get-pip.py | python3.9 \
44
  && update-alternatives --install /usr/bin/python python /usr/bin/python3.9 1 \
45
  && update-alternatives --install /usr/bin/pip pip /usr/local/bin/pip 1 \
46
- && rm -rf /var/lib/apt/lists/*
 
 
 
 
 
 
 
47
 
48
  # Verify Python and pip versions
49
  RUN python --version && pip --version
 
27
 
28
  # Install system dependencies as root
29
  USER root
30
+ # Install basic utilities, Python 3.9, graphical libraries, and add NVIDIA cuDNN repository
31
  RUN apt-get update && apt-get install -y \
32
  wget \
33
  curl \
 
43
  && curl -sS https://bootstrap.pypa.io/get-pip.py | python3.9 \
44
  && update-alternatives --install /usr/bin/python python /usr/bin/python3.9 1 \
45
  && update-alternatives --install /usr/bin/pip pip /usr/local/bin/pip 1 \
46
+ && apt-get install -y gnupg && \
47
+ curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub | gpg --dearmor -o /usr/share/keyrings/cuda-archive-keyring.gpg && \
48
+ echo "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /" > /etc/apt/sources.list.d/cuda.list && \
49
+ apt-get update && apt-get install -y libcudnn8=8.9.3.* libcudnn8-dev=8.9.3.* && \
50
+ rm -rf /var/lib/apt/lists/*
51
+
52
+ # Verify cuDNN version
53
+ RUN dpkg-query -W libcudnn8
54
 
55
  # Verify Python and pip versions
56
  RUN python --version && pip --version