GPU 版本训练,程序刚启动时,可能会出现下面的错误信息:
Could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED
error retrieving driver version: Unimplemented: kernel reported driver version not implemented on Windows
解决方案:
我在网上查了很多办法都说是cudnn版本问题,我下载的是cuda9.0和对应的cudnn。
然后我不断的换版本也不能解决问题,最后发现是tensorflow-cpu和tensorflow-gpu不能同时存在,不然就会引发错误,所以我创建了一个新的虚拟环境然后只安装了tensorflow-gpu就完美解决了(浪费了我一个下午~结果居然这么简单就解决了)。不过gpu速度比cpu确实快了很多。
希望能够帮到大家~~~