一颗行走的大白菜:
问题描述:
本人系统ubuntu 16.04 + cuda 8.0 + cudnn 5.1 在安装tensorflow 的过程中遇到的问题是 :
我选择virtualenv的安装方式,用的是python 2.7 版本,安装很简单,也很顺利,可是,蛋壳是,在验证tensorflow 是否安装成功时报错,首先报
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
这个错误是说,需要我安装cuda 9.0 版本,可是我不想升级安装,于是我采用tensorflow 降级安装,
首先删除 rm -rf ~/tensorflow
之前安装的tensorflow 版本,
然后安装 pip install tensorflow-gpu==1.4
, 安装完成后再去测试,然后发现还有错,如下
ImportError: libcublas.so.6.0: cannot open shared object file: No such > file or directory
这次报6.0的错,原因在于我的cudnn是5.1 人家需要6.0 于是,不想再往下降级安装tensorflow,我下载了cudnn6.0
安装方式如下:
tar -xzvf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
然后问题解决
希望能帮到你