Spaces:
Configuration error
Configuration error
Kangarroar
commited on
Commit
•
7ba04d9
1
Parent(s):
2dea9f5
Update Dockerfile
Browse files- Dockerfile +48 -30
Dockerfile
CHANGED
@@ -1,18 +1,44 @@
|
|
1 |
-
|
|
|
|
|
|
|
2 |
|
3 |
FROM base as base-amd64
|
4 |
|
5 |
-
ENV
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
|
7 |
-
ENV
|
8 |
-
ENV
|
9 |
-
ENV NV_CUDA_COMPAT_PACKAGE cuda-compat-11-8
|
10 |
|
|
|
|
|
|
|
|
|
11 |
FROM base as base-arm64
|
12 |
|
13 |
-
ENV
|
14 |
-
ENV
|
15 |
-
ENV
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
|
17 |
FROM base-${TARGETARCH}
|
18 |
|
@@ -21,30 +47,22 @@ ARG TARGETARCH
|
|
21 |
LABEL maintainer "NVIDIA CORPORATION <[email protected]>"
|
22 |
|
23 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
28 |
&& rm -rf /var/lib/apt/lists/*
|
29 |
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
RUN apt-get update && apt-get install -y --no-install-recommends \
|
34 |
-
cuda-cudart-11-8=${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 |
-
# nvidia-container-runtime
|
46 |
-
ENV NVIDIA_VISIBLE_DEVICES all
|
47 |
-
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
|
48 |
|
49 |
|
50 |
WORKDIR /app
|
|
|
1 |
+
ARG IMAGE_NAME
|
2 |
+
FROM ${IMAGE_NAME}:11.8.0-runtime-ubuntu20.04 as base
|
3 |
+
|
4 |
+
ENV NV_CUDA_LIB_VERSION "11.8.0-1"
|
5 |
|
6 |
FROM base as base-amd64
|
7 |
|
8 |
+
ENV NV_CUDA_CUDART_DEV_VERSION 11.8.89-1
|
9 |
+
ENV NV_NVML_DEV_VERSION 11.8.86-1
|
10 |
+
ENV NV_LIBCUSPARSE_DEV_VERSION 11.7.5.86-1
|
11 |
+
ENV NV_LIBNPP_DEV_VERSION 11.8.0.86-1
|
12 |
+
ENV NV_LIBNPP_DEV_PACKAGE libnpp-dev-11-8=${NV_LIBNPP_DEV_VERSION}
|
13 |
+
|
14 |
+
ENV NV_LIBCUBLAS_DEV_VERSION 11.11.3.6-1
|
15 |
+
ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME libcublas-dev-11-8
|
16 |
+
ENV NV_LIBCUBLAS_DEV_PACKAGE ${NV_LIBCUBLAS_DEV_PACKAGE_NAME}=${NV_LIBCUBLAS_DEV_VERSION}
|
17 |
|
18 |
+
ENV NV_NVPROF_VERSION 11.8.87-1
|
19 |
+
ENV NV_NVPROF_DEV_PACKAGE cuda-nvprof-11-8=${NV_NVPROF_VERSION}
|
|
|
20 |
|
21 |
+
ENV NV_LIBNCCL_DEV_PACKAGE_NAME libnccl-dev
|
22 |
+
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION 2.16.2-1
|
23 |
+
ENV NCCL_VERSION 2.16.2-1
|
24 |
+
ENV NV_LIBNCCL_DEV_PACKAGE ${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda11.8
|
25 |
FROM base as base-arm64
|
26 |
|
27 |
+
ENV NV_CUDA_CUDART_DEV_VERSION 11.8.89-1
|
28 |
+
ENV NV_NVML_DEV_VERSION 11.8.86-1
|
29 |
+
ENV NV_LIBCUSPARSE_DEV_VERSION 11.7.5.86-1
|
30 |
+
ENV NV_LIBNPP_DEV_VERSION 11.8.0.86-1
|
31 |
+
ENV NV_LIBNPP_DEV_PACKAGE libnpp-dev-11-8=${NV_LIBNPP_DEV_VERSION}
|
32 |
+
|
33 |
+
ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME libcublas-dev-11-8
|
34 |
+
ENV NV_LIBCUBLAS_DEV_VERSION 11.11.3.6-1
|
35 |
+
ENV NV_LIBCUBLAS_DEV_PACKAGE ${NV_LIBCUBLAS_DEV_PACKAGE_NAME}=${NV_LIBCUBLAS_DEV_VERSION}
|
36 |
+
|
37 |
+
ENV NV_LIBNCCL_DEV_PACKAGE_NAME libnccl-dev
|
38 |
+
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION 2.16.2-1
|
39 |
+
ENV NCCL_VERSION 2.16.2-1
|
40 |
+
ENV NV_LIBNCCL_DEV_PACKAGE ${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda11.8
|
41 |
+
|
42 |
|
43 |
FROM base-${TARGETARCH}
|
44 |
|
|
|
47 |
LABEL maintainer "NVIDIA CORPORATION <[email protected]>"
|
48 |
|
49 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|
50 |
+
libtinfo5 libncursesw5 \
|
51 |
+
cuda-cudart-dev-11-8=${NV_CUDA_CUDART_DEV_VERSION} \
|
52 |
+
cuda-command-line-tools-11-8=${NV_CUDA_LIB_VERSION} \
|
53 |
+
cuda-minimal-build-11-8=${NV_CUDA_LIB_VERSION} \
|
54 |
+
cuda-libraries-dev-11-8=${NV_CUDA_LIB_VERSION} \
|
55 |
+
cuda-nvml-dev-11-8=${NV_NVML_DEV_VERSION} \
|
56 |
+
${NV_NVPROF_DEV_PACKAGE} \
|
57 |
+
${NV_LIBNPP_DEV_PACKAGE} \
|
58 |
+
libcusparse-dev-11-8=${NV_LIBCUSPARSE_DEV_VERSION} \
|
59 |
+
${NV_LIBCUBLAS_DEV_PACKAGE} \
|
60 |
+
${NV_LIBNCCL_DEV_PACKAGE} \
|
61 |
&& rm -rf /var/lib/apt/lists/*
|
62 |
|
63 |
+
# Keep apt from auto upgrading the cublas and nccl packages. See https://gitlab.com/nvidia/container-images/cuda/-/issues/88
|
64 |
+
RUN apt-mark hold ${NV_LIBCUBLAS_DEV_PACKAGE_NAME} ${NV_LIBNCCL_DEV_PACKAGE_NAME}
|
65 |
+
ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
66 |
|
67 |
|
68 |
WORKDIR /app
|