在安装Anaconda后安装tensorflow GPU版本,运行:
pip3 install tensorflow-gpu
然后进入python环境测试
python
import tensorflow
然后出现了问题:
报错关键在这里:ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
于是在官网https://www.tensorflow.org/install/install_linux#common_installation_problems上找到了这是安装tensorflow的常见报错,解决办法:
本人的cudnn和cuda是实验室配置没问题,所以考虑是tensorflow安装的问题,于是找到这个解决办法:
即再安装tensorflow CPU版本即可:
pip3 install tensorflow
运行个小例子:
成功!