python中怎么查看cuda是否可用
这篇文章主要介绍“python中怎么查看cuda是否可用”,在日常操作中,相信很多人在python中怎么查看cuda是否可用问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”python中怎么查看cuda是否可用”的疑惑有所帮助!接下来,请跟着小编一起来学习吧!
创新互联是一家专业提供讷河企业网站建设,专注与成都网站制作、成都做网站、HTML5建站、小程序制作等业务。10年已为讷河众多企业、政府机构等服务。创新互联专业网站制作公司优惠进行中。
具体操作方法:
1、在电脑开始菜单中找到python程序。
2、在python程序中打开idle工具。
3、在idle工具中点击左上角的“File”并选择“New File”新建一个shell脚本。
4、在shell脚本中输入以下指令导入torch模块。
import torch #导入torch模块
5、最后通过以下指令查看cuda是否可用即可。
print(torch.cuda.is_available())
运行结果示例:
>>>print(torch.cuda.is_available())
True #返回True说明cuda可用
相关操作指令:
print(torch.cuda.device_count()) #查看可行的cuda数目
到此,关于“python中怎么查看cuda是否可用”的学习就结束了,希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习,快去试试吧!若想继续学习更多相关知识,请继续关注创新互联网站,小编会继续努力为大家带来更多实用的文章!
标题名称:python中怎么查看cuda是否可用
URL网址:http://scyanting.com/article/iispij.html