diff --git a/Dockerfile b/Dockerfile index 528ed9ad7cbaae930e3e7976075d2255ef882967..374e00bc9823ad068df9ccdfb60a2153290b0c3a 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.8.0-base-ubuntu22.04 +FROM nvidia/cuda:12.2.0-base-ubuntu22.04 ARG VERSION ENV VERSION=1.6.0