Skip to content

Commit 57a7631

Browse files
authored
fix: update cuda env vars
1 parent b525f97 commit 57a7631

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ ARG CUDA_VERSION=11.8
44
ARG CUDNN_VERSION=8.9.4.25
55
ARG TENSORRT_VERSION=8.6.1.6
66
FROM python:$PYTHON_VERSION-slim AS base
7+
ENV CUDA_VERSION=11.8
78
ENV CUDNN_VERSION=8.9.4.25
89
ENV TENSORRT_VERSION=8.6.1.6
910
# Install wget.

0 commit comments

Comments
 (0)