killawhale/apex repository overview

This image contains Pytorch 1.2 with Python 3.6 and CUDA 10.0. It also has NVIDIA Apex pre-installed for mixed precision training and an array of common packages used in deep learning research.