ubuntu 16.04安装tensorflow-GPU
安装tensorflow-GPU 版本需要先安装英伟达显卡驱动和cuda, cudnn。
一定要确保三者的版本相互匹配
各版本匹配关系如下图:
安装pip3:
sudo apt-get install python3-pip python3-dev
更新pip3:
python 3.x版本:sudo pip3 install --upgrade pip
安装tensorflow:
sudo pip3 install tensorflow-gpu==1.5 -i https://pypi.tuna.tsinghua.edu.cn/simple
安装后进行测试:
注意,对ubuntu16.04来说,默认python是2.7,这里打开。.bashrc文件,在文件底部加入alias python = python3
在python中输入import tensorflow as tf,显示如下问题:
ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory
按照如下链接方法可以解决。
https://blog.****.net/weixin_40298200/article/details/79420758