帮助中心/最新通知

质量为本、客户为根、勇于拼搏、务实创新

< 返回文章列表

【科技前沿】windows系统下如何确认CUDA和CUDNN都安装成功了

发表时间:2025-01-16 01:32:56 小编:主机乐-Yutio

前言

本文概述:在Windows操作系统下,很多人只知道怎么判断CUDA是否安装成功,却不知道该怎么样去判断CUDNN也安装成功,因此作者特地出了一篇文章来作为教程。 作者介绍:作者本人是一名人工智能炼丹师,目前在实验室主要研究的方向为生成式模型,对其它方向也略有了解,希望能够在CSDN这个平台上与同样爱好人工智能的小伙伴交流分享,一起进步。谢谢大家鸭~~~

如果你觉得这篇文章对您有帮助,麻烦点赞、收藏或者评论一下,这是对作者工作的肯定和鼓励。

确认CUDA安装成功

进入如下CUDA目录

启动cmd命令行节目并将deviceQuery拖入命令行窗口, 然后按回车运行

出现这个,则说明第一项测试通过

然后同理将bandwidthTest拖入命令行并回车

出现这个结果,则第二项测试也通过了 。有些网传说第二项测试是检查cudnn是否正常的这是错误的,因为不安装cudnn这项测试也可以通过。通过这两项测试则CUDA安装正确了。

确认CUDNN安装是否成功

目前博主知道的确认CUDNN安装是否成功的办法需要已经安装好pytorch框架才能确认

打开命令行,然后进入你已经安装好pytorch的conda环境,然后输入python并按回车

然后依次执行下列的命令

代码语言:javascript
AI代码解释
复制
>>> 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安装正确。


联系我们
返回顶部