我先从镜像安装gpu版本的tensorflow:
sudo pip install -i https://pypi.douban.com/simple tensorflow-gpu
之后,在python下import tensorflow出错:
- ImportError: libcublas.so.9.0: cannot open shared object file: No
such file or directory
这是因为tensorflow与cuda之间有版本对应关系,我安装的是cuda-10,最新(截至2019.02)的tensorflow-1.11.0只支持cuda-9,所以会找不到libcublas.so.9.0。
版本对应关系:https://blog.csdn.net/omodao1/article/details/83241074
所以重装cuda-9.0:(可以不卸载cuda-10.0,但要改环境变量)
从官网下载cuda-9.0的deb包
sudo dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda-9.0
sudo gedit ~/.bashrc
把~/.bashrc中的
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/targets/x86_64-linux/lib
中的cuda-10.0改成cuda-9.0
再source ~/.bashrc
即可
此时再import tensorflow就不会报错了。