我爱编程

tensorflow安装

2018-01-03  本文已影响0人  sweetsky0901

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

上一篇下一篇

猜你喜欢

热点阅读