diff --git a/gpu.Dockerfile b/gpu.Dockerfile index e37c76a7..1c0a8db6 100644 --- a/gpu.Dockerfile +++ b/gpu.Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.2.0-devel-ubuntu20.04 AS builder +FROM nvidia/cuda:11.2.2-devel-ubuntu20.04 AS builder # Env variables ENV DEBIAN_FRONTEND=noninteractive \ @@ -21,7 +21,7 @@ RUN bash configure.sh clean ### Use a second image for the final asset to reduce the number and # size of the layers. -FROM nvidia/cuda:11.2.0-runtime-ubuntu20.04 +FROM nvidia/cuda:11.2.2-runtime-ubuntu20.04 #FROM nvidia/cuda:11.2.0-devel-ubuntu20.04 # Env variables