HE ye (Alex)

Official Docker images for the machine learning framework TensorFlow (http://www.tensorflow.org)

nvidia/cuda

CUDA and cuDNN images from gitlab.com/nvidia/cuda

Official Caffe images

PyTorch is a deep learning framework that puts Python first.

GPU-enabled Keras using TensorFlow/ CNTK/ MXNET/ Theano backend.

A docker image with caffe and the Xilinx Deephi dnndk v2.08 installed in a Ubuntu 16.04 image

Docker images for Darknet Yolo v4, v3 & v2. (https://github.com/AlexeyAB/darknet)

The PyTorch on ROCm Docker images