rheartpython/cvdeep_horovod repository overview

Nvidia GPU-enabled with cudnn. Exposes a port with Jupyterhub for multitenant Python coding in a browser. For running on Linux OS with Nvidia GPU hardware. Utlizes Horovod as well. More updates coming as this image gets developed.

Specs: Base image: nvidia/cuda:9.0-devel-ubuntu16.04

ENV TENSORFLOW_VERSION=1.8.0 ENV PYTORCH_VERSION=0.4.0 ENV CUDNN_VERSION=7.0.5.15-1+cuda9.0 ENV NCCL_VERSION=2.2.12-1+cuda9.0

Usage:
sudo nvidia-docker run -it -v /var/run/docker.sock:/var/run/docker.sock -p 8788:8788 --expose=8788 rheartpython/cvdeep_horovod