Skip to content

n1lov/docker-ml

Repository files navigation

Docker for ML/DL


Requirements

Nvidia graphics card with CUDA support http://www.geforce.com/hardware/technology/cuda/

Nvidia driver http://www.geforce.com/drivers

Docker https://www.docker.com/get-docker

Docker Compose https://docs.docker.com/compose/install/


Installation

Nvidia Modprobe sudo apt-get install nvidia-modprobe

Run:

install-nvidia-docker.sh

install-nvidia-docker-compose.sh


Images

8.0-cudnn5-devel-ubuntu16.04

tensorflow 0.12.1-gpu-py3


Create images and test gpu support

tensorflow_gpu_test.sh

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published