Fix: update cudnn env variables (#8)

superlinear-ai · Sep 7, 2023 · b525f97 · b525f97
1 parent 41791b9
commit b525f97
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -4,7 +4,8 @@ ARG CUDA_VERSION=11.8
 ARG CUDNN_VERSION=8.9.4.25
 ARG TENSORRT_VERSION=8.6.1.6
 FROM python:$PYTHON_VERSION-slim AS base
-
+ENV CUDNN_VERSION=8.9.4.25
+ENV TENSORRT_VERSION=8.6.1.6
 # Install wget.
 RUN apt-get update && \
     apt-get install --yes wget && \