This is a continuation of my previous post on setting up a home PC for deep learning.
Below are some software installation instructions which might help you to get started.
The OS of choice for deep learning is Linux. And of many Linux distributions, the easiest one to get started is probably Ubuntu. Hence, I’ve installed desktop version of Ubuntu 16.06. Installing Ubuntu in Virtual Machine will not make sense if you plan on using GPU, so it has to be “real” Ubuntu installation.
Go to GeForce Drivers and select your particular OS and GPU card. In my case - GTX 1080 and Linux 64. The driver I got (as of 2016-09-29) has version 367.44.
Installing the driver.
Ctrl+Alt+F3and then login with your credentials
- Stop lightdm service like this:
$ sudo service lightdm stop
- Start driver installer. (When prompted, I skipped installing 32 bit compatibility)
$ sudo ~/Downloads/NVIDIA-Linux-x86_64-367.44.run
- After reboot, you can check that the driver is installed and working if everything looks pretty, and the following command finds your card and shows correct driver version:
Go to CUDA HOMEPAGE and download the latest (at the time of this writing 8.0) version of CUDA for your architecture and distribution. I chose “runfile” as installer type.
- Once download is done, do:
$ sudo sh ~/Downloads/cuda_8.0.44_linux.run
When asked, you can choose to skip driver installation in case you’ve installed it during previous step. If you do chose to install driver, the installation might fail if you did not stop lightdm before.
~/.bashrcto include the following at the end:
export PATH=/usr/local/cuda-8.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64/$LD_LIBRARY_PATH
$ source ~/.bashrc
- To verify that it worked type
To get cuDNN, you’ll need to register in Accelerated Computing Developer Program. It is free, and once you’ve registered, you can download it. cuDNN. You need to get version 5.1 for Linux and CUDA 8.0.
- Once download is done, do:
$ tar -xvf cudnn-8.0-linux-x64-v5.1.tgz $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 $ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
(That’s it - it is just few files which need to be downloaded and copied to appropriate location)
- Install some Python libraries
$ sudo apt-get update $ sudo apt-get upgrade $ sudo apt-get install python-pip $ sudo apt-get install python-numpy swig python-dev python-wheel
- Installing Bazel - open-source version of Google’s build system
$ sudo add-apt-repository ppa:webupd8team/java $ sudo apt-get update $ sudo apt-get install oracle-java8-installer $ echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list $ curl https://storage.googleapis.com/bazel-apt/doc/apt-key.pub.gpg | sudo apt-key add - $ sudo apt-get update && sudo apt-get install bazel $ sudo apt-get upgrade bazel
- Install Git and clone Tensorflow’s repository
$ sudo apt-get install git $ sudo clone https://github.com/tensorflow/tensorflow
- Configure by going to tensorflow’s folder and running
Here I configured without Google cloud and Hadoop Filesystem support but with GPU support. Specify CUDA’s version as 8.0 when asked and for GTX 1080, compute capability of 6.0
- Run build (this will take some time - in my case it took 1031 seconds!):
$ bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
Technically, master branch might be broken or just not work with your configuration for whatever reason. Try switching to more stable “version” branch instead.
- Setup for development
$ mkdir _python_build $ cd _python_build $ ln -s ../bazel-bin/tensorflow/tools/pip_package/build_pip_package.runfiles/org_tensorflow/* . $ ln -s ../tensorflow/tools/pip_package/* . $ sudo python setup.py develop
Now you should be able to run some models. Test it with MNIST by going to
$ python convolutional.py. It’ll first download some data and then should run really fast. You can check using
nvidia-smithat your GPU usage goes up.