运行TensorFlow代码时报错

运行TensorFlow代码时报错

错误信息ImportError: libcublas.so.10.0: cannot open shared object file

原因：TensorFlow版本与CUDA版本不匹配

可使用pip3 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