為了調研
https://github.com/tensorflow/models/tree/master/research/object_detection
安裝tensorflow在GPU的機器上
pip instrall tensorflow-gpu
遇到的最大的問題是會報各種關于cudnn和cuda的錯誤
總結一下滞磺,其實需要安裝環(huán)境中的cuda和cudnn的版本盘寡,和編譯安裝的tensorflow的版本是一致的滤港。才可以
比如開始環(huán)境里裝了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