【本地环境】
操作系统:Window server 2012 R2 64bits
NVIDIA版本:Tesla K80,driver版本用nvidia-smi命令查看是376.51
这样的情况下自己安装tensorflow-gpu时太大意了,直接conda install tensorflow-gpu,结果自动安装了 9.0 版本的cudatoolkit 和7.1.4版本的cudnn,然后程序就报了错误(忘记截图了):CUDA driver version is insufficient for CUDA runtime version
原因就是cudatoolkit 9.0 和 nvidia driver 376.51 不匹配
把cuda和tensorflow-gpu降级成下属版本即可:
Cudatoolkit版本:8.0
cudnn版本:6.0
Tensorflow-gpu版本:1.1.0
当然也可以升级nvidia driver 至和 cudatoolkit匹配的版本
具体nvidia driver匹配哪个版本 cudatoolkit 可以在nvidia官网查到download diver
参考链接如下:
https://blog.csdn.net/yourgreatfather/article/details/79645062