初学PyTorch

1.导入pytorch并查看是否可用

import torch
print(torch.cuda.is_available())

2.查看可用设备

print(torch.cuda.device_count())
print(torch.cuda.current_device())
print(torch.cuda.get_device_name(0))

3.打开torch

查看具体用法

print(dir(torch))
dir(torch.cuda)
dir(torch.cuda.is_available)
help(torch.cuda.is_available)