pytorch GPU操作事例

>>> import torch 
>>> if_cuda =torch.cuda.is_available()
>>> print("if_cuda=",if_cuda)
if_cuda= True
>>> gpu_count = torch.cuda.device_count()
>>> print("gpu_count=",gpu_count)
gpu_count= 8