Spaces:
Configuration error
Configuration error
Kangarroar
commited on
Commit
•
6cce418
1
Parent(s):
21a4885
Update Dockerfile
Browse files- Dockerfile +7 -7
Dockerfile
CHANGED
@@ -4,15 +4,15 @@ FROM base as base-amd64
|
|
4 |
|
5 |
ENV NVARCH x86_64
|
6 |
|
7 |
-
ENV NVIDIA_REQUIRE_CUDA "cuda>=
|
8 |
-
ENV NV_CUDA_CUDART_VERSION
|
9 |
-
ENV NV_CUDA_COMPAT_PACKAGE cuda-compat-
|
10 |
|
11 |
FROM base as base-arm64
|
12 |
|
13 |
ENV NVARCH sbsa
|
14 |
-
ENV NVIDIA_REQUIRE_CUDA "cuda>=
|
15 |
-
ENV NV_CUDA_CUDART_VERSION
|
16 |
|
17 |
FROM base-${TARGETARCH}
|
18 |
|
@@ -27,11 +27,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
27 |
apt-get purge --autoremove -y curl \
|
28 |
&& rm -rf /var/lib/apt/lists/*
|
29 |
|
30 |
-
ENV CUDA_VERSION
|
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-
|
35 |
${NV_CUDA_COMPAT_PACKAGE} \
|
36 |
&& rm -rf /var/lib/apt/lists/*
|
37 |
|
|
|
4 |
|
5 |
ENV NVARCH x86_64
|
6 |
|
7 |
+
ENV NVIDIA_REQUIRE_CUDA "cuda>=11.8 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"
|
8 |
+
ENV NV_CUDA_CUDART_VERSION 11.8.89-1
|
9 |
+
ENV NV_CUDA_COMPAT_PACKAGE cuda-compat-11-8
|
10 |
|
11 |
FROM base as base-arm64
|
12 |
|
13 |
ENV NVARCH sbsa
|
14 |
+
ENV NVIDIA_REQUIRE_CUDA "cuda>=11.8"
|
15 |
+
ENV NV_CUDA_CUDART_VERSION 11.8.89-1
|
16 |
|
17 |
FROM base-${TARGETARCH}
|
18 |
|
|
|
27 |
apt-get purge --autoremove -y curl \
|
28 |
&& rm -rf /var/lib/apt/lists/*
|
29 |
|
30 |
+
ENV CUDA_VERSION 11.8.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-11-8=${NV_CUDA_CUDART_VERSION} \
|
35 |
${NV_CUDA_COMPAT_PACKAGE} \
|
36 |
&& rm -rf /var/lib/apt/lists/*
|
37 |
|