Install the latest TensorFlow version that supports CUDA, cudNN, and GPU on a Windows 11 PC

introduce

If you're anything like me, you know that setting up TensorFlow on a new device can be a pain. In this tutorial, I'll cover installing CUDA, cuDNN, and TensorFlow on Windows 11. The goal of this article is to make that process as painless as possible. This article assumes that you already have a CUDA-compatible GPU installed on your PC.

installation steps

It is important to follow each step in order. If you do the steps out of order, you'll have to start the process all over again, which no one wants to do. So, without further ado, let's get straight to the point.

Software we need to install:

Microsoft Visual Studio (2019 Community Edition)
NVIDIA CUDA Toolkit
NVIDIA cuDNN
Python (I personally use anaconda)
TensorFlow

Step 1: Remove all current NVIDIA drivers

Go to the search bar on Windows and search for "Add or Remove Programs". You should now see a list of installed applications. Scroll down until you see anything that says NVIDIA. It should look similar to the image below.

insert image description here
Now check and remove anything marked NVIDIA on it. Don't worry about removing them, we'll install them back in a later step.

Step 2: Install Visual Studio (not VS code)

Installing the CUDA toolkit requires Visual Studio. If you try to unsecure

Guess you like

Origin blog.csdn.net/iCloudEnd/article/details/132423014