Question : get gpu name tensorflow and pytorch
Answered by : merwan
# tensorflow
from tensorflow.python.client import device_lib
devices_tf = device_lib.list_local_devices()
devices_tf = print(devices)
# pytorch
import torch
devices_torch = torch.cuda.get_device_name()
print(devices_torch)Source : | Last Update : Thu, 02 Dec 21