Ubuntu20.04确认cuda和cudnn已经安装成功

当我们通过官网安装cuda和cudnn时,终端执行完命令后我们仍不能确定是否已经安装成功。接下来教大家用几句命令测试。

  • cuda

检测版本号

nvcc -V

如果输出如下,则安装成功。
可以看到版本号是11.4
**

  • cudnn

**
检测版本号
有两种命令:如果你的cudnn版本较旧,就用第一条。否则就用第二条。(当然你也可以两条都试一试,当安装成功时才会有输出。

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

正确输出如图
在这里插入图片描述
谢谢观看。

版权声明:本文为CSDN博主「济善245」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_55749979/article/details/122746800

济善245

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

win下conda安装多版本cuda

需求/问题 电脑只能安装一个cuda,但是往往研究tensorflow-gpu的时候,可能需要多个版本的cuda,利用conda可以实现安装多个版本。 安装教程 查看当前cuda版本 nvcc

pt模型在VS上运行

注意:本例子为我个人对YOLOv5的源码进行在VS上的实现,大家可按自己的模型根据源码进行修改 我这为单个物体的检测,如果检测多个物体需对结果处理函数(non_max_suppression2)进行修改