How to Setup TensorFlow GPU 2.2 with NVIDIA GPUs

Hello world, it’s Aaron!

Image Copyright TensorFlow.
Image Copyright NVIDIA.

Install CUDA

lspci | grep -i nvidia
uname -m && cat /etc/*release

Install Anaconda and TensorFlow GPU

conda create --name tf-gpu
conda activate tf-gpu
pip install tensorflow-gpu==2.2.0rc2
python3
import tensorflow as tf
tf.__version__ # Result should be '2.2.0-rc2'
tf.config.list_physical_devices('GPU') # should list all available GPUs

-Aaron Ma https://aaronhma.com

--

--

Planet Earth, The Milky Way, Local Group, Virgo Supercluster, Laniakea Supercluster, the Universe

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
Aaron Ma

Planet Earth, The Milky Way, Local Group, Virgo Supercluster, Laniakea Supercluster, the Universe