在深度学习跑论文代码的时候,安装好环境后,经常会验证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

在这里插入图片描述

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐