上面是我出现的问题,查了一下说是TensorFlow的版本和CUDA的版本不一致造成的问题。
首先可以先分别查一下TensorFlow和CUDA的版本号。
TensorFlow版本查询
pip list | grep tensorflow
我的结果是这样的:
CUDA版本查询
cat /usr/local/cuda/version.txt
我的结果是这样的:
CUDNN查询
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
我的结果:
所以我这里的TensorFlow版本和CUDA的版本是对应的,版本号对应请自行百度。如果版本号对应,还出现上述问题。我是这样解决的。
运行如下代码:
ldconfig /usr/local/cuda-10.0/lib64
我的问题就解决了。