Skip to content

Instantly share code, notes, and snippets.

@johnny-brav0
johnny-brav0 / Install_Tensforflow2_with_GPU_in_Ubuntu20.04.md
Last active March 4, 2023 13:12 — forked from raulqf/Install_Tensforflow2_with_GPU_in_Ubuntu20.04.md
How to install Tensorflow2 with GPU (cuDNN and tensorRT) support in Ubuntu 20.04

How to install Tensorflow2 with GPU (cuDNN and tensorRT) support in Ubuntu 20.04

Installation of Tensorflow 2 with GPU support is easy, there are just a couple things to check before proceeding:

  • CUDA compatibility with your Nvidia Graphics Card. Please check here for your hardware version.

  • Tensorflow also lists a compatibility table. It is advised to install compatible Tensorflow-CUDA-cuDNN versions, otherwise it may lead to issues related to versions of shared libraries in tensorflow.

  • CUDA version also requires a minimum Nvidia driver version which can be checked in the following CUDA COMPATIBILITY TABLE.

NOTE: To install locally in a conda environment check the last section of this post.