Skip to Content
Install TensorFlow-GPU on Windows 10: cuDNN, CUDA toolkit, and Visual Studio for Application Development
on-demand course

Install TensorFlow-GPU on Windows 10: cuDNN, CUDA toolkit, and Visual Studio for Application Development

with Kishan Kumar
November 2020
Beginner
43m
English
Apress
Closed Captioning available in English

Overview

Skip wasting time while trying to get TensorFlow-GPU up and running by stepping through the complex procedure to learn what steps are pivotal and which aren’t. Gain a basic overview of TensorFlow-GPU and why it might be the right choice for your machine learning and deep learning development environment. Then look at which version of Python will fit best for your needs and how to get it to interact properly with your TF-GPU. You’ll also find out how to gauge whether or not your graphics card is well suited to the task and what your options are based on your hardware. 

Once all the basic requirements are met, we’ll install the Cuda toolkit to provide a development environment for creating high-performance GPU-accelerated applications. The toolkit includes GPU-accelerated libraries, debugging and optimization tools, a C/C++ compiler, and a runtime library to deploy your applications.

We’ll also need to install Visual Studio IDE for the C++ development libraries that will be required by the toolkit. Many users miss this step and they run into the problem of their toolkit not installing properly.

Having done all these steps we will then look into cuDNN, which is a deep neural network library. This library provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, normalization, and activation layers. We’ll have to provide a path to cuDNN, too, as it’s not an application but a library. So we’ll look into how to make our system locate these libraries. 

Finally, we’ll install TensorFlow-GPU. We’ll verify it by running some basic commands and also verify whether it’s making use of your GPU or not. TensorFlow-GPU offers a powerful, hardware-dependent development environment for the most consumptive of deep learning tasks.

What You Will Learn

  • Install the packages needed for TensorFlow-GPU
  • Set a path in Windows 10
  • Use the correct version of Python for your needs

Who This Video Is For

Developers new to deep learning who would like to use the more powerful GPU for hardware intensive deep-learning applications.

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.

Watch now

Unlock full access

More than 5,000 organizations count on O’Reilly

AirBnbBlueOriginElectronic ArtsHomeDepotNasdaqRakutenTata Consultancy Services

QuotationMarkO’Reilly covers everything we've got, with content to help us build a world-class technology community, upgrade the capabilities and competencies of our teams, and improve overall team performance as well as their engagement.
Julian F.
Head of Cybersecurity
QuotationMarkI wanted to learn C and C++, but it didn't click for me until I picked up an O'Reilly book. When I went on the O’Reilly platform, I was astonished to find all the books there, plus live events and sandboxes so you could play around with the technology.
Addison B.
Field Engineer
QuotationMarkI’ve been on the O’Reilly platform for more than eight years. I use a couple of learning platforms, but I'm on O'Reilly more than anybody else. When you're there, you start learning. I'm never disappointed.
Amir M.
Data Platform Tech Lead
QuotationMarkI'm always learning. So when I got on to O'Reilly, I was like a kid in a candy store. There are playlists. There are answers. There's on-demand training. It's worth its weight in gold, in terms of what it allows me to do.
Mark W.
Embedded Software Engineer

You might also like

Deep Belief Nets in C++ and CUDA C: Volume 3: Convolutional Nets

Deep Belief Nets in C++ and CUDA C: Volume 3: Convolutional Nets

Timothy Masters

Publisher Resources

ISBN: 9781484266984