tensorflow与cuda不得不说的秘密(坑)

这两天配置tensorflow实在是搞得精疲力竭,之前一直以为是tensorflow版本的问题,后来在网上查阅相关资料后才知道是这个问题

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

出现以上问题的原因在于cuda的驱动版本与cuda的runtime版本不一致,cuda的驱动版本可以通过

nvidia-smi

来查看而驱动对应的cuda版本可以在网上查到,所以在安装tensorflow前,可以先用anaconda安装cuda驱动版本对应的cuda,再装tensorflow,目前实验室的cuda版本应该装9.0

 conda install cuda==9.0

在安装好对应的cuda版本后,直接安装tensorflow就会自动安装对应已有cuda版本的tensorflow了

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值