Skip to content

Instantly share code, notes, and snippets.

View gaberogan's full-sized avatar
😃
Smiling

Gabriel Rogan gaberogan

😃
Smiling
View GitHub Profile
@gaberogan
gaberogan / Install_Tensforflow2_with_GPU_in_Ubuntu20.04.md
Created February 22, 2022 17:35 — forked from raulqf/Install_Tensforflow2_with_GPU_in_Ubuntu20.04.md
How to install Tensorflow2 with GPU support in Ubuntu 20.04

How to install Tensorflow2 with GPU support in Ubuntu 20.04

Installation of Tensorflow2 with GPU support is easy and the only complication can be arisen from the CUDA compability which in turns depends on the Nvidia driver version. Before going farther, please check if your Nvidia Video Card is compatible with the required versions that are defined in this gist, use this link.

Tensorflow offers in its website a table of the compatibility between libraries for the target OS. You can visit that website in the following COMPATIBILITY TABLE that points to the Tensorflow installation from source for linux. For the time writing this gist, Tensorflow2 v2.5.0 requires CUDA v11.2 and CUDNN v8.2. It is really important to match the exact version, otherwise tensorflow will have problems loading the shared libraries as not finding the correct version.

CUDA version also requires for a minimum Nvidia driver version