diff --git a/devops/base/Dockerfile b/devops/base/Dockerfile index 491ccf46a..9b470cbb7 100644 --- a/devops/base/Dockerfile +++ b/devops/base/Dockerfile @@ -1,4 +1,4 @@ -ARG WBIA_UBUNTU_IMAGE=nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04 +ARG WBIA_UBUNTU_IMAGE=nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 #nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 #nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04 #nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04