在ubuntu上安装tensorflow-gpu版本
pip3 install tensorflow-gpu
安装成功后,运行
import tensorflow
出现错误
ImportError:libcublas.so.9.0: cannot open shared object file: No such file or directory
这是因为默认安装的tensorflow-gpu是1.5版本的,需要cuda-9.0支持。但是之前安装的是cuda-7.5,所以重新安装tensorflow-gpu。
pip3 install tensorflow-gpu==1.3