Kangarroar commited on
Commit
6cce418
1 Parent(s): 21a4885

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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>=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
 
@@ -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 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
 
 
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