起因:更新tensorflow。
运行代码时便报错:
ImportError: libcublas.so.10.0: cannot open shared object file
查找原因发现是tensorflow的版本和cuda的版本不匹配。
可使用
pip install tensorflow-gpu==[version]
将tensorflow版本切换到指定版本,其中,对应的版本号[version]可在Tensorflow官网查看。
另:
1.查看cuda版本(在ubuntu下):
cat /usr/local/cuda/version.txt
或者
nvcc --version
2.查看cudnn版本(在ubuntu下):
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
3.查看tensorflow-gpu的版本:
pip list