dl env

Check PyTorch GPU

$ torch.cuda.is_available()

Check TensorFlow GPU

$ import tensorflow as tf

$ tf.config.list_physical_devices('GPU')