Spaces:
Configuration error
Configuration error
Kangarroar
commited on
Commit
•
21a4885
1
Parent(s):
5eb3109
Update Dockerfile
Browse files- Dockerfile +50 -1
Dockerfile
CHANGED
@@ -1,4 +1,53 @@
|
|
1 |
-
FROM
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
|
3 |
WORKDIR /app
|
4 |
|
|
|
1 |
+
FROM ubuntu:20.04 as base
|
2 |
+
|
3 |
+
FROM base as base-amd64
|
4 |
+
|
5 |
+
ENV NVARCH x86_64
|
6 |
+
|
7 |
+
ENV NVIDIA_REQUIRE_CUDA "cuda>=12.0 brand=tesla,driver>=450,driver<451 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=510,driver<511 brand=unknown,driver>=510,driver<511 brand=nvidia,driver>=510,driver<511 brand=nvidiartx,driver>=510,driver<511 brand=geforce,driver>=510,driver<511 brand=geforcertx,driver>=510,driver<511 brand=quadro,driver>=510,driver<511 brand=quadrortx,driver>=510,driver<511 brand=titan,driver>=510,driver<511 brand=titanrtx,driver>=510,driver<511 brand=tesla,driver>=515,driver<516 brand=unknown,driver>=515,driver<516 brand=nvidia,driver>=515,driver<516 brand=nvidiartx,driver>=515,driver<516 brand=geforce,driver>=515,driver<516 brand=geforcertx,driver>=515,driver<516 brand=quadro,driver>=515,driver<516 brand=quadrortx,driver>=515,driver<516 brand=titan,driver>=515,driver<516 brand=titanrtx,driver>=515,driver<516 brand=tesla,driver>=520,driver<521 brand=unknown,driver>=520,driver<521 brand=nvidia,driver>=520,driver<521 brand=nvidiartx,driver>=520,driver<521 brand=geforce,driver>=520,driver<521 brand=geforcertx,driver>=520,driver<521 brand=quadro,driver>=520,driver<521 brand=quadrortx,driver>=520,driver<521 brand=titan,driver>=520,driver<521 brand=titanrtx,driver>=520,driver<521"
|
8 |
+
ENV NV_CUDA_CUDART_VERSION 12.0.107-1
|
9 |
+
ENV NV_CUDA_COMPAT_PACKAGE cuda-compat-12-0
|
10 |
+
|
11 |
+
FROM base as base-arm64
|
12 |
+
|
13 |
+
ENV NVARCH sbsa
|
14 |
+
ENV NVIDIA_REQUIRE_CUDA "cuda>=12.0"
|
15 |
+
ENV NV_CUDA_CUDART_VERSION 12.0.107-1
|
16 |
+
|
17 |
+
FROM base-${TARGETARCH}
|
18 |
+
|
19 |
+
ARG TARGETARCH
|
20 |
+
|
21 |
+
LABEL maintainer "NVIDIA CORPORATION <[email protected]>"
|
22 |
+
|
23 |
+
RUN apt-get update && apt-get install -y --no-install-recommends \
|
24 |
+
gnupg2 curl ca-certificates && \
|
25 |
+
curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${NVARCH}/3bf863cc.pub | apt-key add - && \
|
26 |
+
echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${NVARCH} /" > /etc/apt/sources.list.d/cuda.list && \
|
27 |
+
apt-get purge --autoremove -y curl \
|
28 |
+
&& rm -rf /var/lib/apt/lists/*
|
29 |
+
|
30 |
+
ENV CUDA_VERSION 12.0.0
|
31 |
+
|
32 |
+
# For libraries in the cuda-compat-* package: https://docs.nvidia.com/cuda/eula/index.html#attachment-a
|
33 |
+
RUN apt-get update && apt-get install -y --no-install-recommends \
|
34 |
+
cuda-cudart-12-0=${NV_CUDA_CUDART_VERSION} \
|
35 |
+
${NV_CUDA_COMPAT_PACKAGE} \
|
36 |
+
&& rm -rf /var/lib/apt/lists/*
|
37 |
+
|
38 |
+
# Required for nvidia-docker v1
|
39 |
+
RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf \
|
40 |
+
&& echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
|
41 |
+
|
42 |
+
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
|
43 |
+
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
|
44 |
+
|
45 |
+
COPY NGC-DL-CONTAINER-LICENSE /
|
46 |
+
|
47 |
+
# nvidia-container-runtime
|
48 |
+
ENV NVIDIA_VISIBLE_DEVICES all
|
49 |
+
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
|
50 |
+
|
51 |
|
52 |
WORKDIR /app
|
53 |
|