美文网首页
failed call to cuDevicePrimaryCt

failed call to cuDevicePrimaryCt

作者: 我的章鱼小丸子呢 | 来源:发表于2020-07-24 11:36 被阅读0次
    报错信息:
    错误
    原因:

    利用CPU版本的tensorflow时调用了cuda,可能是因为服务器行装了cuda及cudnn

    解决方案:

    在脚本的开头添加如下代码

    import os
    os.environ["CUDA_VISIBLE_DEVICES"]="-1"
    

    相关文章

      网友评论

          本文标题:failed call to cuDevicePrimaryCt

          本文链接:https://www.haomeiwen.com/subject/wqrjlktx.html