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