Pytorch查看版本 

import torch

# 查看版本
print(torch.__version__)
# 查看GPU是否可用
print(torch.cuda.is_available())
# 查看GPU型号
if torch.cuda.is_available():
    print(torch.cuda.get_device_name())
posted @ 2023-12-13 23:40  碧水云天4  阅读(111)  评论(0)    收藏  举报