File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 1
1
# syntax=docker/dockerfile:1
2
- ARG DEBIAN_FRONTEND=noninteractive
3
2
ARG PYTHON_VERSION=3.8
4
3
FROM python:$PYTHON_VERSION-slim AS base
5
4
5
+ ARG DEBIAN_FRONTEND=noninteractive
6
6
ARG CUDA=11.8
7
7
ARG NV_CUDA_CUDART_VERSION=11.8.89-1
8
8
ARG NV_CUDA_COMPAT_PACKAGE=cuda-compat-11-8
@@ -14,7 +14,7 @@ ARG LIBINVER_VERSION=8.4.3-1+cuda11.6
14
14
# NVIDIA: https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.8.0/ubuntu2204/base/Dockerfile
15
15
# specify the version of the CUDA Toolkit to use and the which driver versions are compatible for each brand of GPU.
16
16
17
- # ENV NVARCH x86_64
17
+ ENV NVARCH x86_64
18
18
ENV CUDA $CUDA
19
19
ENV CUDNN_VERSION $CUDNN_VERSION
20
20
ENV NV_CUDA_CUDART_VERSION $NV_CUDA_CUDART_VERSION
You can’t perform that action at this time.
0 commit comments