在深度学习跑论文代码的时候,安装好环境后,经常会验证torch的版本、以及torch与cuda版本是否对应、cuda是否可用、以及torch对应的cuda的版本。代码如下!
import torch
print(torch.__version__)
print(torch.cuda.is_available())
结果如图:
也可查看可行的cuda数目:
print(torch.cuda.device_count())
查看torch对应的cuda版本:
torch.version.cuda