diff --git a/devops/base/Dockerfile b/devops/base/Dockerfile index 8c87fcc94..c9c66e349 100644 --- a/devops/base/Dockerfile +++ b/devops/base/Dockerfile @@ -1,4 +1,5 @@ -ARG WBIA_UBUNTU_IMAGE=nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 +ARG WBIA_UBUNTU_IMAGE=nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 +#nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 #nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04 FROM ${WBIA_UBUNTU_IMAGE} as org.wildme.wbia.base