1. 前提说明
假设已经装好了 Ubuntu+python3.4+pip3+virtual environment+tensorflow 1.8.0 。本文主要收集 tensorflow 安装过程中的 bugs
2. Import error: libcublas.so.9.0: cannot open shared object file: no such file or directory
分析:缺少 libcublas.so.9.0。大概率怀疑是版本不对。
解决方法:
(1)到 cuda 库目录( 我的是 /usr/local/cuda/lib64,这个因人而异,可以通过 which nvcc 查看 cuda 路径)
(2)发现只有 libcublas.so.8.0。于是网上查找 tensorflow 1.4.1 支持 libcublas.so.8.0 ,于是安装 tensorflow-gpu 1.4.1
3. Import error: libcudnn.so.6 cannot open shared object: no such file or directory
(1) 到 cuda 库目录,发现只有 libcudnn.so.5
(2) 有了上面的经验,立即在网上查找,得知 tensorflow 1.2 支持 libcudnn.so.5,于是安装 tensorflow-gpu 1.2
命令: pip install tensorflow-gpu==1.2
4. 验证:
(1)python
(2) import tensorflow as tf
成功!(喜极而泣脸)