遇到的问题:
代码要求CUDA9.0+tensorflow1.4.0,但是用的服务器没有权限,不能改任何东西,被迫无奈,选择了合适的版本解决冲突。
一、这个问题是你的cuda跟tensorflow版本冲突问题
(nvcc -V 查看cuda版本
cudnn:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
)可以看一下对应版本
二、我用的服务器cuda10.0.130,没有权限等等问题,只能改tensorflow(建议建一个自己的环境)
直接复制
pip install tensorflow-gpu==1.14(版本随便改,对应上面的)
三、运行代码完美解决。