为了调研
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
网友评论