tensorflow-安装

 

1、pip安装(最好在虚拟环境中安装)

→更新pip:pip install --upgrade pip

→安装最新版tensorflow(GPU):pip install tensorflow-gpu

    安装指定版本tensorflow(GPU):pip install tensorflow-gpu==版本号(例如:pip install tensorflow-gpu==1.2)

 

可能遇见的问题

ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory(import tensorflow时)

原因1:安装了tensorflow1.3或更高的版本,1.3以下的版本可以使用cuDNN v5.1,而1.3及以上版本需要cuDNN v6.0做支持了

解决方案:(1)安装tensorflow1.2或更低版本;(2)安装cuDNN v6.0

 

posted on 2017-12-20 14:19  执剑长老  阅读(197)  评论(0编辑  收藏  举报