Public | Automated Build

Last pushed: a year ago
Short Description
CUDA7.5 + CUDNN3 on CentOS7, with common build tools.
Full Description

Docker images for Deep Learning

This repository aims at delivering trouble-free docker images for people
wanting to try out deep learning frameworks.

See the respective README files to learn how to start and use them.

docker-nvidia is the base image, providing an install of cuda 7.5 on centos7,
with CUDNN4 and common tools for compiling a framework.

You can build the images with the Makefiles provided:

  • from the top directory :
    make caffe
  • from the projects' directory:
    cd docker-caffe; make

If you have trouble accessing the GPUs from inside the Docker containers, make
sure you have matching versions of the CUDA drivers.

Your own version (as provided by yum info cuda-drivers) should match the
number in
docker-nvidia/Dockerfile.
If necessary, change the version in the Dockerfile and rebuild the images.

Docker Pull Command
Owner
nitnelave
Source Repository

Comments (0)