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.