pytorch学习——检查torch是否可用，是否与CUDA匹配，服务器上有几张卡可用

'''
用于检查torch是否可以工作
cuda是否可以与torch匹配
检查一下有几张卡可以用
'''
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.device_count())
cout = torch.cuda.device_count()
for i in range(0,cout):
    print('第',i,'张：',torch.cuda.get_device_name(i))

这样可以检查服务器配置

配合nvidai-smi检查当前服务器显存使用情况