运行TensorFlow代码时报错
错误信息ImportError: libcublas.so.10.0: cannot open shared object file
原因:TensorFlow版本与CUDA版本不匹配
可使用pip3 install tensorflow-gpu==[version]
将TensorFlow版本切换到制定版本中去,其对应的version可在查看
另:
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