Deb packages for major releases (i.e. v3.0 and v4.0 but not v3.1) are provided for easy installation on Ubuntu 14.04. If these packages don't meet your needs, then you can follow these instructions to build DIGITS and its dependencies from source.
You need an NVIDIA driver (details and instructions).
Run the following commands to get access to some package repositories:
# Access to CUDA packages
CUDA_REPO_PKG=cuda-repo-ubuntu1404_7.5-18_amd64.deb
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/${CUDA_REPO_PKG} -O /tmp/${CUDA_REPO_PKG}
sudo dpkg -i /tmp/${CUDA_REPO_PKG}
rm -f /tmp/${CUDA_REPO_PKG}
# Access to Machine Learning packages
ML_REPO_PKG=nvidia-machine-learning-repo-ubuntu1404_4.0-2_amd64.deb
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64/${ML_REPO_PKG} -O /tmp/${ML_REPO_PKG}
sudo dpkg -i /tmp/${ML_REPO_PKG}
rm -f /tmp/${ML_REPO_PKG}
# Download new list of packages
sudo apt-get update
Now that you have access to all the packages you need, installation is simple.
sudo apt-get install digits
Through package dependency chains, this installs digits
, caffe-nv
, torch7-nv
, libcudnn
and many more packages for you automatically.
The DIGITS server should now be running at http://localhost/
.
See comments below if you run into any issues.
Now that you're up and running, check out the Getting Started Guide.
If you have another server running on port 80 already, you may need to reconfigure DIGITS to use a different port.
% sudo dpkg-reconfigure digits
To make other configuration changes, try this (you probably want to leave most options as "unset" or "default" by hitting ENTER
repeatedly):
% cd /usr/share/digits
# set new config
% sudo python -m digits.config.edit -v
# restart server
% sudo stop nvidia-digits-server
% sudo start nvidia-digits-server
If you try to install a new driver while the DIGITS server is running, you'll get an error about CUDA being in use. Shut down the server before installing a driver, and then restart it afterwards:
% sudo stop nvidia-digits-server
# (install driver)
% sudo start nvidia-digits-server
The DIGITS server runs as www-data
, so keep in mind that prebuilt LMDB datasets used for generic models need to be readable by the www-data
user.
In particular, the entire chain of directories from /
to your data must be readable by www-data
.
There is at least one Torch package which is missing a required dependency on cusparse. If you see this error:
/usr/share/lua/5.1/cunn/THCUNN.lua:7: libcusparse.so.7.5: cannot open shared object file: No such file or directory
The simplest fix is to manually install the missing library:
% sudo apt-get install cuda-cusparse-7-5
% sudo ldconfig
If you run into an issue not addressed here, try searching through the GitHub issues and/or the user group.