Ubuntu 16.04下安装CUDA8.0+Caffe+TensorFlow(4)

TensorFlow的安装还是比较简单的。
1、下载对应版本的TensorFlow
2、执行安装 sudo pip install tensorflow_gpu-0.12.0rc1-cp35-cp35m-linux_x86_64.whl
3、在python中输入import tensorflow as tf测试

上面输出的带W的警告，是因为没有通过源代码编译TF，CPU的参数没有开启，因此CPU不支持一系列协议。不过，这只会影响CPU的计算速度，并不影响GPU的计算。
如果是在终端下运行，不显示这个警告。

- 6、在Pycharm中进行配置

1、Caffe配置成功后，在Pycharm中出现ImportError

Pycharm中TensorFlow可以直接import导入。但是在import caffe的时候会出现ImportError: No module named caffe的情况。可以通过手动添加库解决。
a.依次打开File>Setting>Project Interpreter.打开右上角的小齿轮。选择第三个选项 more...

b.选择选择最右侧边框上最后一个图标

c.点击右侧加号，增加路径

d.添加自己的Caffe下的Python路径

e.这样便可以成功导入Caffe测试效果如下

2、如果还有cannot libcudart 的错误

error:如果caffe还不能正常import 出现 cannot libcudart 的错误，那么用如下方法解决。

a.进入pycharm-community-2016.2/bin
b.使用sudo gedit pycharm.sh 打开pycharm.sh文件
c.

#找到这行 LD_LIBRARY_PATH="$IDE_BIN_HOME:$LD_LIBRARY_PATH" "$Java_BIN" \ #添加环境变量 LD_LIBRARY_PATH='/usr/local/cuda/lib64'
# 修改后为 LD_LIBRARY_PATH='/usr/local/cuda/lib64' LD_LIBRARY_PATH="$IDE_BIN_HOME:$LD_LIBRARY_PATH" "$JAVA_BIN" \

d.重新启动pycharm，之前报错的脚本就能正常运行了
最后效果如下所示

Attention:有的命令加上sudo反而会出现错误，遇到错误的时候，可以尝试去掉sudo试一下。

Ubuntu 15.04 下Caffe + + CUDA 7.0 安装配置指南 

Caffe 深度学习入门教程 

Ubuntu 16.04下Matlab2014a+Anaconda2+OpenCV3.1+Caffe安装

Ubuntu 16.04系统下CUDA7.5配置Caffe教程

Caffe在Ubuntu 14.04 64bit 下的安装

深度学习框架Caffe在Ubuntu下编译安装 

Caffe + Ubuntu 14.04 64bit + CUDA 6.5 配置说明 

Ubuntu 16.04上安装Caffe

Caffe配置简明教程 ( Ubuntu 14.04 / CUDA 7.5 / cuDNN 5.1 / OpenCV 3.1 ) 

Ubuntu 16.04安装 Caffe GPU版 

Ubuntu 16.04上安装Caffe(CPU only)