注:此篇为源码编译方式安装Tensorflow,不过也可以通过其他方法,比如Anaconda,我没试过:)
一、参考资料:
【http://blog.csdn.net/zhaoyu106/article/details/52793183/】
【https://sanwen8.cn/p/23d9QiU.html】这个测试Tensorflow的方式和上面那个不一样,我也只是测试”通过源代码编译TensorFlow使其支持GPU的方式”
【http://m.blog.csdn.net/article/details?hmsr=toutiao.io&id=52658965&utm_medium=toutiao.io&utm_source=toutiao.io】
二、事先装好Ubuntu16.04+CUDA8.0
三、cuDNN v5:利用GPU加速卷积神经网络,提高整个网络训练速度。
拷贝库文件:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*配置环境变量:
- 因没有~/.bash_profile文件,改为修改~/.bashrc,添加以下路径:
export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64”
export CUDA_HOME=/usr/local/cuda- 因没有~/.bash_profile文件,改为修改~/.bashrc,添加以下路径:
路径生效:
source ~/.bashrc
四、安装其他库
pip:
sudo apt-get install python-pip python-dev
Bazel:(谷歌开源构建工具,类似于Make,用于构建Tensorflow)
方法一:
貌似不能用,采用第二种方法:
方法二:
五、安装Tensorflow(由于开头找到的资料是关于Tensorflow0.11的,所以下载该版本的Tensorflow,否则安装时会有其他参数)
按照参考博客来就行,期间遇到一个问题:
注意第4步
sudo pip install /home/*(你自己的用户名)/Desktop/tensorflow-0.10.0-cp2-none-any.whl,
我替换成
sudo pip install /tmp/tensorflow_pkg/tensorflow-0.11.0-py2-none-any.whl
出现错误【TypeError: unsupported operand type(s) for -=: ‘Retry’ and ‘int’】。解决办法(估计需要python2,而不是python3):
apt-get remove python-pip python3-pip
sudo apt-get install python-pip python-dev六、按照博客地址http://blog.csdn.net/zhaoyu106/article/details/52793183/测试结果: