Spaces:
Runtime error
Runtime error
Sébastien De Greef
commited on
Commit
·
8bb21dd
1
Parent(s):
e469db9
chore: Update Dockerfile to use CUDA 12.4.1-devel image and set environment variables more efficiently
Browse files- Dockerfile +4 -6
Dockerfile
CHANGED
|
@@ -1,8 +1,6 @@
|
|
| 1 |
-
FROM nvidia/cuda:12.4.1-
|
| 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 && \
|
|
@@ -31,8 +29,8 @@ RUN rm -f /etc/apt/sources.list.d/*.list && \
|
|
| 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 |
-
|
| 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 \
|
|
|
|
| 1 |
+
FROM nvidia/cuda:12.4.1-devel-ubuntu22.04
|
| 2 |
|
| 3 |
+
ENV DEBIAN_FRONTEND=noninteractive TZ=Europe/Paris TERM=linux
|
|
|
|
|
|
|
| 4 |
|
| 5 |
# Install some basic utilities
|
| 6 |
RUN rm -f /etc/apt/sources.list.d/*.list && \
|
|
|
|
| 29 |
|
| 30 |
ENV PATH="/home/chatbot/.local/bin:${PATH}"
|
| 31 |
|
| 32 |
+
#RUN wget -q https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_550.54.15_linux.run
|
| 33 |
+
# \ && sudo /bin/bash cuda_12.4.1_550.54.15_linux.run
|
| 34 |
|
| 35 |
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 \
|
| 36 |
&& sudo dpkg -i cudnn-local-repo-ubuntu2204-9.1.1_1.0-1_amd64.deb \
|