Update Dockerfile
Browse files- 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
|
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 |
-
&&
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|