Install cuda and torch on Ubuntu 16.04 or newer - View it on GitHub