import torch
print(torch.cuda.is_available()) # cuda是否可用
print(torch.version.cuda) # 查看cuda版本
print(torch.cuda.device_count()) # 查看GPU数量
print(torch.__version__) # 查看torch版本
print(torch.backends.cudnn.version()) # 检查cudnn版本
12-09
5888
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)
08-13
276
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)