我是在使用anaconda的Spyder写一个数字识别的任务的时候遇到这个问题的,当时我以为是我的函数库有问题,然后安装了很多东西都没用,然后百度的答案都很模糊,在我尝试了一天一下午之后,找到了问题原因所在。
首先我用的keras,它是基于TensorFlow2.0,而TensorFlow2.0基于的显卡驱动版本是cuda10.0,cuda10.0需要的Driver Version必须大于411.31。
1、在桌面右键打开NVIDIA面板
2、找到自己的driver版本
3、若是发现自己与下图的版本匹配不一样,TensorFlow2.0必须基于cuda10.0
3、先升级自己的driver版本,去英伟达官网https://www.nvidia.cn/
(1)
(2)在这里找到自己的显卡,要是不知道的话2、步骤有显卡版本
(3)一般需要安装最新的版本,安装过程为傻瓜式安装,安装完这个之后,就需要安装cuda了,下面是cuda10.0的下载界面,这个也是傻瓜式安装。
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal