tensorflow
cdknight_happy
种一棵树最好的时间是十年前,其次是现在!
展开
-
windows下tensorflow CUDA_ERROR_ILLEGAL_ADDRESS解决办法
最近在使用tensorflow的时候出现了以下所示的bug:2017-11-08 12:24:52.838039: E tensorflow/stream_executor/cuda/cuda_driver.cc:1080] failed to synchronize the stop event: CUDA_ERROR_ILLEGAL_ADDRESS2017-11-08 12:24:52.83...原创 2018-05-09 10:58:07 · 9642 阅读 · 1 评论 -
tensorflow1.8 c++ 加载训练好的模型 踩坑
tensorflow c++下加载训练好的图和checkpoint的代码是:const std::string pathToGraph = "./model/model.ckpt-29520.meta"; const std::string checkpointPath = "./model/model.ckpt-29520"; //gpu option tf::SessionOp...原创 2018-05-11 10:50:44 · 4894 阅读 · 8 评论 -
win10 tensorflow1.8 c++ set_visible_device_list / SetDefaultDevice 问题解决
1.问题描述使用CMake编译tensorflow1.8的c++版本,得到了tensorflow.lib和tensorflow.dll,但是在使用下面所示的方法tensorflow::graph::SetDefaultDevice("/device:GPU:0", &graph_def);或者tensorflow::ConfigProto* config = &am...原创 2018-07-18 16:33:51 · 3066 阅读 · 12 评论