当前位置:   article > 正文

Win10 TensorFlow: failed call to cuInit: CUDA_ERROR_UNKNOWN_failed call to cuinit: cuda_error_unknown: unknown

failed call to cuinit: cuda_error_unknown: unknown error

一、驱动问题

首先,查看一下自己的电脑需要怎样的驱动,我们可以先到 http://www.nvidia.com/Download/index.aspx 查询下我们需要的是怎样的驱动,这里我的显卡是 GTX 1080,所以以此为例说明,勾选好对应的配置:

点击 Search,可以看到查询结果如下所示:

这里说明我们需要的版本是 417.35。

接下来如果我们之前安装了驱动的话,可以重新安装一下,如果当前已经安装好了就不必了。

第一步:GeForce Experience

第二步:更新驱动程序,安装完后,关闭自动更新

二、软件版本问题

如若上一步无法解决问题:

请确定cuda与cudnn版本是匹配的,并且cudnn的版本应对应TensorFlow。

例如:

Cuda 9.2

使用管理员权限安装,一路默认。

安装完成之后打开命令行,输入:nvcc -V查看版本,如果出现以下类似信息表示安装成功。

安装成功后,我的电脑上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V9_2两个环境变量。

 

我们还需要添加如下几个变量:

在用户变量Path下添加:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

设置完成之后,我们可以打开命令行来查看。

 

Tensorflow

目前,tensorflow还不能支持cuda9.2,所以只能通过源码编译或者别人编译好的安装包安装。大神编译的各种版本的tensorflow安装包地址:

https://github.com/fo40225/tensorflow-windows-wheel

下载完成后将

tensorflow-1.9.0-cp36-cp36m-win_amd64.whl

tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64.whl

复制到C:\Windows\System32文件夹下,然后执行命令:

pip install tensorflow-1.9.0-cp36-cp36m-win_amd64.whl

pip install  tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64.whl
--------------------- 
注意:一定不要tensorflow-gpu和tensorflow(cpu版)一起装,因为这样装有个先后顺序问题,先安装tensorflow-gpu再安装tensorflow,gpu版本直接不能用了。

cudnn

从上面,可以看到对应的cudnn7.1,并在官方下载:

https://developer.nvidia.com/rdp/cudnn-archive

经一步一步筛查,问题解决:

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/311493
推荐阅读
相关标签
  

闽ICP备14008679号