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

如何查看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版本