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

Handsome_zhou

1. #查看计算机的硬件条件：
   
2. 
   
3. n_cudas = torch.cuda.device_count()
   
4. for i in range(n_cudas):
   
5.         print(torch.cuda.get_device_name(i))
   
6. 
   
7. #定义硬件：
   
8. 
   
9. device = 'cuda' if torch.cuda.is_available() else 'cpu'

复制代码

学会了

:(:Q:handshake