tensorflow安装

作者: sweetsky0901 | 来源:发表于2018-01-03 12:24 被阅读0次

    为了调研
    https://github.com/tensorflow/models/tree/master/research/object_detection
    安装tensorflow在GPU的机器上
    pip instrall tensorflow-gpu

    遇到的最大的问题是会报各种关于cudnn和cuda的错误
    总结一下,其实需要安装环境中的cuda和cudnn的版本,和编译安装的tensorflow的版本是一致的。才可以
    比如开始环境里装了cudnn6.5,但是实际tensorflow需要的是6.0.那么就会报错。

    ImportError: /home/aip/.jumbo/lib/python2.7/site-packages/tensorflow/python/../libtensorflow_framework.so: undefined symbol: cudnnGetRNNParamsSize

    确定版本可以用
    strings path/libtensorflow_framework.so | grep libcudnn

    相关文章

      网友评论

        本文标题:tensorflow安装

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