关于机器上已安装CUDA,但在anaconda下tensorflow出现cudaGetDevice() failed问题的解决

机器已经安装了CUDA 9.1,驱动为390.58,我补装了对应的cudnn,并且测试通过,但是在anaconda环境下,用conda install tensorflow-gpu=1.12后,运行sess = tf.Session()时报错:

tensorflow.python.framework.errors_impl.InternalError: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version

原因在于conda自动安装时,CUDA toolkit安装的是9.2,但是9.2对应的驱动为不小于396,因此就出错了
解决方法为在那个环境下卸载CUDA与cudnn,然后安装合适的版本。
卸载:

conda uninstall cudnn
conda uninstall cudatoolkit

安装

conda install cudatoolkit=9.0
conda install cudnn

版本根据需求自定义,下面有版本对应关系。
然后重新用conda install安装tensorflow-gpu即可。
CUDA各版本与驱动的对应关系可点击链接:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
在这里插入图片描述
tensorflow-gpu与CUDA和cudnn的对应关系可查看:https://tensorflow.google.cn/install/source
在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值