Advanced Installation
Deep learning involves a huge amount of matrix multiplications, and Graphic Processing Units (GPUs) are a very important aspect when one begins to learn deep learning. Without a GPU, the experiment process may take a day or more. With a good GPU, we can quickly iterate over deep learning networks and large training datasets, and run multiple experiments in a short amount of time. With TensorFlow, we can work on a single GPU or even multiple GPUs with ease. However, most machine learning platform installations are very complicated once GPUs get involved.
In this chapter, we are going to discuss GPUs and focus on a step-by-step CUDA setup and a GPU-based TensorFlow installation. We will start by installing Nvidia drivers, ...
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