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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM