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檢查當前服務器顯存使用情況