diff --git a/Dockerfile b/Dockerfile index 1c44d3a6..2c8b176e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,5 @@ # base image maintained by the NVIDIA CUDA Installer Team - https://hub.docker.com/r/nvidia/cuda/ -FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 +FROM nvidia/12.8.0-cudnn-devel-ubuntu24.04 # install os packages RUN apt-get update \