Deep Learning with TensorFlow - Second Edition
by Giancarlo Zaccone, Vihan Jain, Md. Rezaul Karim, Motaz Saad
The TensorFlow GPU setup
To use TensorFlow with NVIDIA GPUs, the first step is to install the CUDA Toolkit.
Note
To know more, visit https://developer.nvidia.com/cuda-downloads.
Once the CUDA Toolkit is installed, you must download the cuDNN v5.1 library for Linux from https://developer.nvidia.com/cudnn.
cuDNN is a library that helps accelerate deep learning frameworks, such as TensorFlow and Theano. Here's a brief explanation from the NVIDIA website:
"The NVIDIA CUDA® Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. cuDNN provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, normalization, and activation layers. cuDNN is part of the NVIDIA ...
Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.
Read now
Unlock full access