diff --git a/Dockerfile b/Dockerfile
index 59e7d2f..fc7c546 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,3 +8,5 @@ RUN apt update && apt install -y --no-install-recommends \
         sudo \
         nano \
     && rm -rf /var/lib/apt/lists/*
+
+RUN /usr/sbin/update-nvidia-cudnn -u