Python 查看pytorch版本、判断CUDA是否可用
在深度学习跑论文代码的时候,安装好环境后,经常会验证torch的版本、以及torch与cuda版本是否对应、cuda是否可用。代码如下!import torchprint(torch.__version__)print(torch.cuda.is_available())结果如图:...
文章共257字 · 阅读需要大约1分钟
一键AI生成摘要,助你高效阅读
问答
·
在深度学习跑论文代码的时候,安装好环境后,经常会验证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
更多推荐
已为社区贡献2条内容
所有评论(0)