Spaces:
Runtime error
Runtime error
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- 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
|
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
|