tensorflow
文章平均质量分 90
qq_41870658
这个作者很懒,什么都没留下…
展开
-
tensorflow保存的ckpt,meta模型freeze成pb
参考:https://blog.csdn.net/yjl9122/article/details/78341689使用博主方法时遇到两个问题:1.freeze的时候参数没写全会报错。(解决办法:参数最好写全,无论必选还是可选项)2.加载.meta文件的时候,设置input_binary为False会报错。修改free_graph_meta.py文件时,修改位置跟博主的位置行号对应不上。(解决方法:Ctrl+F 检索所有input_graph_def =的行,把所有行都修改了)博主代码如下:impo原创 2021-04-14 16:16:25 · 628 阅读 · 0 评论 -
win10 c++编译TensorFlow1.13.1 GPU版
Ubuntu16.04 CUDA10.2 CUDNN7.6.5 tensorflow1.13.1下训练的网络,封装成pb模型,在win10下调用,需要先编译TensorFlow。一开始尝试编译GPU的版本,踩坑没爬上来。先编译一个CPU版本的试试,回头有机会再搞GPU的。参考:https://blog.csdn.net/atpalain_csdn/article/details/97945536,https://blog.csdn.net/weixin_43868576/article/details/1原创 2021-03-23 10:13:34 · 919 阅读 · 0 评论 -
ValueError: You are trying to load a weight file containing 4 layers into a model with 0 layers.
ValueError: You are trying to load a weight file containing 4 layers into a model with 0 layers.已经训练好模型,并保存为Simcnn.h5文件,模型重新编译并调用的时候报错。出错代码: def model_pre(self): if os.path.exists("ckpt/Simcnn/checkpoint/Simcnn.h5"): Simcnn.model.loa原创 2020-09-29 15:39:27 · 675 阅读 · 0 评论 -
tensorflow-gpu,CUDA与CUDNN版本对应关系
转载自:https://blog.csdn.net/Young__Fan/article/details/88130385当在安装Tensorflow的时候,若我们直接使用以下指令:pip install tensorflow-gpu或:pip install --ignore-installed --upgrade tensorflow-gpu这样的话,安装的会是最新版的ten...转载 2019-06-18 20:30:04 · 634 阅读 · 0 评论 -
ubuntu18.04安装tensorflow-gpu
转载:https://blog.csdn.net/u014670893/article/details/82467384 ubuntu18.04 安装tensorflow-gpu 2018年0...转载 2019-06-20 09:49:44 · 453 阅读 · 0 评论 -
从零开始安装双系统Ubuntu18.04、CUDA、cuDNN、TensorFlow-GPU
转载:https://blog.csdn.net/blgpb/article/details/82048478 从零开始安装双系统Ubuntu18.04、CUDA、cuDNN、TensorFlow-GPU ...原创 2019-06-20 09:55:16 · 510 阅读 · 0 评论 -
NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the lat
2019年06月22日 出现这个问题之后参考了一篇博客https://blog.csdn.net/sinat_23619409/article/detail...转载 2019-06-22 15:24:27 · 15707 阅读 · 5 评论