Handsome_zhou 发表于 2022-3-17 13:27:36

查看计算机的硬件条件并且定义模型使用的硬件

#查看计算机的硬件条件：

n_cudas = torch.cuda.device_count()
for i in range(n_cudas):
        print(torch.cuda.get_device_name(i))

#定义硬件：

device = 'cuda' if torch.cuda.is_available() else 'cpu'

学会了 发表于 2022-3-17 14:31:12

:(:Q:handshake
页: [1]
查看完整版本: 查看计算机的硬件条件并且定义模型使用的硬件