这是一个基于Ubuntu 18.04 的Docker镜像,包含了OpenMPI 3.1.2, CUDA 10.2 和 cuDNN 8。
docker.io/leoxiao/openmpi3.1.2-cuda10.2-cudnn8-ubuntu18.04:pt1.8.1-lts