python中怎么查看cuda是否可用

这篇文章主要介绍“python中怎么查看cuda是否可用”，在日常操作中，相信很多人在python中怎么查看cuda是否可用问题上存在疑惑，小编查阅了各式资料，整理出简单好用的操作方法，希望对大家解答”python中怎么查看cuda是否可用”的疑惑有所帮助！接下来，请跟着小编一起来学习吧！

具体操作方法：

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是否可用”的学习就结束了，希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习，快去试试吧！若想继续学习更多相关知识，请继续关注亿速云网站，小编会继续努力为大家带来更多实用的文章！