python中如何查看cuda版本 Python查看cuda版本

主机教程 建站分享 3个月前 (09-16) 35次浏览

文章摘要:python中如何查看cuda版本 Python查看cuda版本

python中查看cuda版本的方法:1、在电脑中找到python程序;2、打开idle工具;3、在idle中 […]

python中查看cuda版本的方法:1、在电脑中找到python程序;2、打开idle工具;3、在idle中新建一个shell脚本;4、输入“import torch”指令导入torch模块;5、通过“print(torch.version.cuda)”查看指令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版本 Python查看cuda版本
文章链接:https://www.7966.org/post/10895.html
转载请注明出处

喜欢 (0)