首页 > 其他 > 详细

查看Tensorflow-gpu对应的CUDA版本和cudnn版本的问题

时间:2020-11-18 21:31:17      阅读:177      评论:0      收藏:0      [点我收藏+]

如何查看tensorflow-gpu对应的CUDA版本和cudnn版本问题(预先安装好Anaconda的前提下)

在路径..\Anaconda3\Lib\site-packages\tensorflow\python\platform下可以看到build_info.py文件

技术分享图片

 

 

技术分享图片

 

 

这是在tensorflow1.4.0中的build_info.py文件,可以看出其对应的CUDA为8.0,cudnn为6.0。

tensorflow1.4以上的版本都可以通过这种方法查看对应的CUDA和cudnn版本。

在tensorflow1.3版本中这种方法不适用,因为没有build_info.py这个文件(亲测)

 

在目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA下可以看到安装的CUDA版本

 

查看Tensorflow-gpu对应的CUDA版本和cudnn版本的问题

原文:https://www.cnblogs.com/elitphil/p/14001832.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!