记录如何在Ubuntu下同时安装tensorflow和caffe

安装opencv 显卡驱动等

这一过程和Ubuntu下安装caffe是一样的不做详细解释,具体过程可参看这一链接中安装opencv和Nvidia显卡驱动的部分.

安装cuda9.0和cudnn7

下载cuda9.0安装包,记住,不要下载错误了,下载成cuda9.1了.执行以下命名

sudo chmod 777 cuda_9.0*.sh
sudo sh cuda_9.0*.run

出现以下提示的时候,需要注意,因为之前安装了驱动,所以这里不要选择安装,其余的都直接默认或者选择是即可。

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 3**.**

完成安装后,编辑~/.bashrc加入以下内容.

export  PATH=/usr/local/cuda9.0/bin${PATH:+:${PATH}}
export  LD_LIBRARY_PATH=/usr/local/cuda9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

并将.bashrc文件生效

source ~/.bashrc

测试cuda

cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery 
make 
sudo ./deviceQuery

安装cudnn7

sudo tar -zxvf ./cudnn-9.0-linux-x64-v7.1.tgz
# 把刚刚解压的那个cuda文件更名为cuda-9.0
mv cuda cuda-9.0

然后执行:

cd cuda-9.0
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/ 
# 建议进行如下操作
cd cuda-8.0
sudo cp lib64/lib* /usr/local/cuda-9.0/lib64/
sudo cp include/cudnn.h /usr/local/cuda-9.0/include/ 

更新软连接:

cd /usr/local/cuda-9.0/lib64/
sudo chmod +r libcudnn.so.7.1.2
sudo ln -sf libcudnn.so.7.1.2 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
sudo ldconfig /usr/local/cuda-9.0/lib64/

# 安装tensorflow

pip install tensorflow-gpu==1.6

测试

python
import tensorflow as tf

报错

如遇报错:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

则对cuda-9的lib64目录执行更新软连接的操作

sudo ldconfig /usr/local/cuda-9.0/lib64/

以上操作不起作用的话，进行以下操作试试：

cd ~
sudo cp /usr/local/cuda-9.0/lib64/libcudart.so.9.0 /usr/local/lib/libcudart.so.9.0 && sudo ldconfig
cp /usr/local/cuda-9.0/lib64/libcublas.so.9.0 /usr/local/lib/libcublas.so.9.0 && sudo ldconfig
cp /usr/local/cuda-9.0/lib64/libcurand.so.9.0 /usr/local/lib/libcurand.so.9.0 && sudo ldconfig

如遇报错:

ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory

则说明cudnn版本有问题或者其导致cudnn未安装成功