Skip to content

ashishpatel26/tensorflow-installation-on-windows10-CUDA-and-cudnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 

Repository files navigation

Tensorflow 2 Installation On Windows10 with CUDA and cudnn

image-20210712165459686


  1. Install anaconda : https://www.anaconda.com/products/individual

  2. Install visual C++ : https://support.microsoft.com/en-us/topic/the-latest-supported-visual-c-downloads-2647da03-1eea-4433-9aff-95f26a218cc0

  3. Check the Link https://www.tensorflow.org/install/source_windows for cuda version

    GPU

    Version Python version Compiler Build tools cuDNN CUDA
    tensorflow_gpu-2.5.0 3.6-3.9 MSVC 2019 Bazel 3.7.2 8.1 11.2
  4. Create the new environment in Conda

Commands :
  1. conda create -n tf_gpu python==3.8
  2. conda activate tf_gpu
  3. conda install cudatoolkit=11.2 cudnn=8.1 -c=conda-forge
  4. pip install --upgrade tensorflow-gpu
  5. Testing GPU is available or Not.

import tensorflow as tf
tf.test.is_gpu_available()
  1. Installation of CUDA Toolkit as well - Download CudaToolkit

    Thanks for Reading