csquare default deep learning image

This Docker image is meant to be use as default runtime on the csquare platform.

This image comes with two flavours:

• csquareai/ml-default:single (or csquareai/ml-default:latest) to train models on a single node
• csquareai/ml-default:parallel which uses OpenMPI and optionally Horovod to train models on multiple nodes

See the list of available tags for more information.

Included software

Basic programs

• cmake
• curl
• git
• g++
• vim
• wget

If you need additional software, please open an issue!

Training frameworks/tools

See the Dockerfile for more details.

Software	Version
TensorFlow	2.3.0
PyTorch	1.6.0
torchvision	2.3.0
cuDNN	7.6.5.32-1+cuda10.1
NCCL	2.7.8-1+cuda10.1
MXNet	1.6.0.post0