diff --git a/docker/Dockerfile b/docker/Dockerfile index c6dbaac..3659601 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 +FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04@sha256:8f9dd0d09d3ad3900357a1cf7f887888b5b74056636cd6ef03c160c3cd4b1d95 RUN echo 'debconf debconf/frontend select Noninteractive' | debconf-set-selections diff --git a/docker/cu116.Dockerfile b/docker/cu116.Dockerfile index f68e4bd..752abe1 100644 --- a/docker/cu116.Dockerfile +++ b/docker/cu116.Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.6.2-devel-ubuntu20.04 +FROM nvidia/cuda:11.8.0-devel-ubuntu20.04@sha256:03681bbd11fea044ff3e3a5d65e190a6f935af30c56e03e740b27e4563f61e0f # Set environment variables ENV NVENCODE_CFLAGS "-I/usr/local/cuda/include"