前言
如果你觉得这篇文章对你有帮助,别忘了点赞、收藏或留下评论,这将是作者持续创作的最大动力和支持。
进入CUDA安装目录
打开cmd命令行工具,将deviceQuery.exe文件拖入窗口,按下回车执行
若出现类似结果,表示第一项设备检测已通过
接着,同样将bandwidthTest.exe拖入命令行并回车运行
Visual Studio IntelliCode
微软VS平台的 AI 辅助开发工具
46 查看详情

随后依次在Python环境中运行以下代码:
>>> import torch>>> print(torch.backends.cudnn.version())# 正常输出应为8200>>> from torch.backends import cudnn # 若无报错则说明导入成功>>> cudnn.is_available() # 正常情况下返回True>>> a = torch.tensor(1.)>>> cudnn.is_acceptable(a.cuda()) # 正常应返回True
如图所示输出结果,表明CUDNN已正确安装并可被PyTorch调用。
以上就是windows系统下如何确认CUDA和CUDNN都安装成功了的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/875348.html
微信扫一扫
支付宝扫一扫