DIGITS recommends a build of Torch7 to use as an alternative backend to Caffe, though it is not required.
For best performance, you'll want:
- One or more NVIDIA GPUs (details)
- An NVIDIA driver (details and installation instructions)
- A CUDA toolkit (details and installation instructions)
- cuDNN (download page)
Install some dependencies with Deb packages:
sudo apt-get install --no-install-recommends git software-properties-common
These instructions are based on the official Torch instructions.
# example location - can be customized
export TORCH_ROOT=~/torch
git clone https://github.com/torch/distro.git $TORCH_ROOT --recursive
cd $TORCH_ROOT
./install-deps
./install.sh -b
source ~/.bashrc
DIGITS requires you to install a few extra dependencies.
sudo apt-get install --no-install-recommends libhdf5-serial-dev liblmdb-dev
luarocks install tds
luarocks install "https://raw.github.com/deepmind/torch-hdf5/master/hdf5-0-0.rockspec"
luarocks install lightningmdb 0.9.18.1-1 LMDB_INCDIR=/usr/include LMDB_LIBDIR=/usr/lib/x86_64-linux-gnu
luarocks install "https://raw.github.com/Sravan2j/lua-pb/master/lua-pb-scm-0.rockspec"
# If you have installed NCCL
luarocks install "https://raw.githubusercontent.com/ngimel/nccl.torch/master/nccl-scm-1.rockspec"
Follow these instructions for information on getting started with Torch7 in DIGITS.