Step by step guide to setup Tensorflow with GPU support on windows 10

Steps:

Step 1: NVIDIA Graphics Driver Installation

Step 2: Visual Studio 2019 Installation

Step 3: Anaconda Installation

Step 4: CUDA Toolkit

CUDA download

Step 5: CUDNN Library

Step 6: Create a Conda environment

conda create --name gpu python=3.8
conda activate gpu
pip install --upgrade tensorflow-gpu
code
import sys
import tensorflow.keras
import tensorflow as tf
import numpy as np
print(f"Tensor Flow Version: {tf.__version__}")
print(f"Keras Version: {tensorflow.keras.__version__}")
print()
print(f"Python {sys.version}")
print("GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
tf.test.is_built_with_cuda()
tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)

--

--

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store