How to install CUDA in Google Colab GPU’s

Question

Cuda is not showing on your notebook because you have not enabled GPU in Colab.

The Google Colab comes with both options GPU or without GPU.
You can enable or disable GPU in runtime settings

Go to Menu > Runtime > Change runtime.

Change hardware acceleration to GPU.

GPU Settings Screenshot

To check if GPU is running or not, run the following command

!nvidia-smi

If the output is like the following image it means your GPU and cuda are working. You can see the CUDA version also. cuda confirmation screenshot

After that to check if PyTorch is capable of using GPU, run the following code.

import torch
torch.cuda.is_available()
# Output would be True if Pytorch is using GPU otherwise it would be False.

To check if TensorFlow is capable of using GPU, run the following code.

import tensorflow as tf
tf.test.gpu_device_name()
# Standard output is '/device:GPU:0'