Tensorflow on Windows 10 with GPU Support – Part 3

This is the final post on getting your windows 10 desktop/laptop ready with Tensorflow (GPU Enabled). The main take away here is that after you install the NVIDIA CUDA SDK, you must install cuDNN version 8.2.2 for CUDA 11.4. Make sure that you select the version of cuDNN that supports your CUDA install.

Download cuDNN

Login or create a developer account at https://developer.nvidia.com/. Once you log in, go to NVIDIA Developer resources. The link to download cuDNN follows:

https://developer.nvidia.com/rdp/cudnn-download and click on the link for your system.

NVIDIA has done a great job with their documentation, so I am going to point you there so that you get the installation guide.

https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#installwindows

Optional – Compile the samples and run them.

Congratulations! Your system is ready to run Tensorflow. I have one more post which has python code that validates if Tensorflow can use the GPU. Stay tunned and happy Tensorflowing…

Leave a comment