Template for machine learning and deep learning projects.
- 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/
- Nvidia Modprobe
sudo apt-get install nvidia-modprobe
Run:
Check docker volume ls
and run nvidia-docker run nvidia/cuda nvidia-smi
if driver volume is empty
tensorflow 0.12.1-gpu-py3