diff --git a/devops/base/Dockerfile b/devops/base/Dockerfile index 8101bc2be..a9e47e6b2 100644 --- a/devops/base/Dockerfile +++ b/devops/base/Dockerfile @@ -1,6 +1,6 @@ -ARG WBIA_UBUNTU_IMAGE=12.1.0-cudnn8-runtime-ubuntu22.04 +ARG WBIA_UBUNTU_IMAGE=nvidia/12.1.0-cudnn8-runtime-ubuntu22.04 #nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 -docker pull nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 +#docker pull 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