tensorflow安装

为了调研
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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。