Spaces:
Configuration error
Configuration error
Kangarroar
commited on
Commit
•
feb8bd8
1
Parent(s):
7ba04d9
Update Dockerfile
Browse files- Dockerfile +1 -67
Dockerfile
CHANGED
@@ -1,70 +1,4 @@
|
|
1 |
-
|
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 |
-
|
45 |
-
ARG TARGETARCH
|
46 |
-
|
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
|
69 |
|
70 |
COPY ./requirements.txt /app/requirements.txt
|
|
|
1 |
+
FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
WORKDIR /app
|
3 |
|
4 |
COPY ./requirements.txt /app/requirements.txt
|