robomakery / caffe-cookbook

Chef cookbook for installing the Caffe deep learning framework

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

caffe-cookbook

Chef cookbook for installing the Caffe deep learning framework.

It was developed using an AWS g2.2xlarge instance running Ubuntu 14.04 (base Ubuntu AMI used was: ami-4ae27e22).

It installs CUDA, cuDNN (see notes below), Caffe, and Caffe's python bindings.

cuDNN

In order to have the cookbook compile Caffe with cuDNN support, you need to download the cuDNN tarball from the NVIDIA website.

You first need to create a CUDA Registered Developer account, wait for the account to be activated, and then download the cuDNN tarball. Once you have the tarball (cudnn-6.5-linux-R1.tgz as of Dec 2014), you should place it in the files/default/cudnn-tarball directory. This will trigger the cookbook to install it and use it when compiling Caffe.

About

Chef cookbook for installing the Caffe deep learning framework


Languages

Language:Ruby 100.0%