How to use system GPU in Jupyter notebook?

Question

I tried a lot of things before I could finally figure out this approach. There are a lot of videos and blogs asking to install the Cuda toolkit and cuDNN from the website. Checking the compatible version. But this is not required anymore all you have to do is the following

pip install tensorflow-gpu

pip install cuda

pip install cudnn

then use the following code to check if your GPU is active in the current notebook

print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

tf.config.list_physical_devices('GPU')

from tensorflow.python.client import device_lib

device_lib.list_local_devices()

tf.test.is_built_with_cuda()

tf.debugging.set_log_device_placement(True)

I just want to confirm, if these steps are enough to enable GPU in jupyter notebook or am I missing something here?

Answer 1

If you installed the compatible versions of CUDA and cuDNN (relative to your GPU), Tensorflow should use that since you installed tensorflow-gpu . If you want to be sure, run a simple demo and check out the usage on the task manager.