温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

python中怎么查看cuda版本

发布时间:2022-05-09 16:58:57 来源:亿速云 阅读:1023 作者:iii 栏目:大数据

这篇文章主要介绍“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.version.cuda)


相关操作指令:

print(torch.cuda.is_available()) #查看cuda是否可用

print(torch.cuda.device_count()) #查看可行的cuda数目


关于“python中怎么查看cuda版本”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识,可以关注亿速云行业资讯频道,小编每天都会为大家更新不同的知识点。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI