Installing TensorFlow 2.3.0 and Torch 1.6.0 on Windows 10 with CUDA Support

Hello World!

Today we will learn to setup TensorFlow 2.3.0 and Torch 1.6.0 on Windows 10 along with CUDA 10.1, as quickly as possible. I assume that you already have Anaconda x64 installed with Python 3.8. So let’s begin.

Download and install CUDA v10.1.243_426.00 from here.

Download and install MS VC Redist x64 from here.

Download and extract cuDNN v7.6.5.32 from here. Copy all files inside this zip file and replace them in the path:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

Open Anaconda Prompt as administrator and run the following command to install TensorFlow GPU.

pip install --upgrade tensorflow-gpu==2.3.0

Run the following command to install Torch.

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

That's all! More details at: