Ubantu16.04, Cuda9.1安装Tensorflow

1 准备

安装Ubantu16.04, Cuda9.x, Cudnn7.0.5, 具体安装方法略

2 安装

参考github安装Tensorflow，在mkl安装的最后有如下命令行

echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib' >> ~/.bashrc

需要将上面的/usr/local/lib修改为自己安装的mkl的so库的位置（可能有多个位置）

问题

在安装成功后，运行Tensorflow测试程序时，如果遇到cannot share cuda 9.1.so，可以尝试将如下指令

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64/

复制到~/.bashrc，然后执行

source ~/.bashrc