这篇文章主要介绍“python中怎么查看cuda版本”的相关知识,小编通过实际案例向大家展示操作过程,操作方法简单快捷,实用性强,希望这篇“python中怎么查看cuda版本”文章能帮助大家解决问题。
我们提供的服务有:成都做网站、网站设计、微信公众号开发、网站优化、网站认证、湖北ssl等。为1000多家企事业单位解决了网站和推广的问题。提供周到的售前咨询和贴心的售后服务,是有科学管理、有技术的湖北网站制作公司
具体操作方法:
1、在电脑开始菜单中找到python程序。
2、在python程序中打开idle工具。
3、在idle工具中点击左上角的“File”并选择“New File”新建一个shell脚本。
4、在shell脚本中输入以下指令导入torch模块。
import torch #导入torch模块
5、最后通过以下指令查看cuda版本即可。
print(torch.version.cuda)
相关操作指令:
print(torch.cuda.is_available()) #查看cuda是否可用
print(torch.cuda.device_count()) #查看可行的cuda数目
关于“python中怎么查看cuda版本”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识,可以关注创新互联行业资讯频道,小编每天都会为大家更新不同的知识点。