首页 > 其他 > 详细

TensorFlow问题

时间:2018-11-19 20:59:31      阅读:291      评论:0      收藏:0      [点我收藏+]

一、查看版本:

進入到Python的命令行狀態後,可以在终端输入查询命令如下:

import tensorflow as tf
tensorflow.__version__

查询tensorflow安装路径为:

tensorflow.__path__

一、cuDNN無法使用

Loaded runtime CuDNN library: 7.0.5 but source was compiled with: 7.1.4.  CuDNN library major and minor version needs to match or have higher minor version in case of CuDNN 7.0 or later version. If using a binary install, upgrade your CuDNN library.  If building from sources, make sure the library loaded at runtime is compatible with the version specified during compile configuration.

我的開發環境

ubuntu16.04   
cuda9.0   
cudnn7.0  
tensorflow 1.12.0  
python 3.6

原因;TensorFlow的版本太高了。

解決方法

pip3 install --upgrade --force-reinstall tensorflow-gpu==1.9.0

 

TensorFlow问题

原文:https://www.cnblogs.com/darklights/p/9985412.html

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