Ubuntu + python + Tensorflow 安装

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
  成功!(喜极而泣脸)

猜你喜欢

转载自blog.csdn.net/yangguangqizhi/article/details/80485116