tensorflow
文章平均质量分 73
li57681522
在职工作中
展开
-
tensorflow的CUDA driver version is insufficient for CUDA runtime version 问题解决方案
CUDA driver version is insufficient for CUDA runtime version 翻译过来就是CUDA的驱动程序版本跟CUDA的运行时版本不匹配!1.CUDA driver version(驱动版本):就是NVIDIA GPU的驱动程序版本;查看命令:nvidia-smi我们看到我的GPU的驱动程序版本是:384.812.CUDA run...原创 2018-09-07 10:25:26 · 49785 阅读 · 19 评论 -
在TensorFlow框架中增加自定义Hook
本笔记演示的是怎么在tensorflow/python/training/basic_session_run_hooks.py增加一个自定义Hook;在其他地方增加Hook可能有所不同,但是基本流程相同,也可参考此笔记。 一.自定义Hook:@tf_export("train.MyHook")class MyHook(session_run_hook.SessionRunHook...原创 2019-02-25 15:22:03 · 2932 阅读 · 1 评论 -
TensorFlow同步训练原理
一、同步训练例子:#coding=utf-8#python sync_dist_train.py --job_name=ps --task_index=0 --issync=1#python sync_dist_train.py --job_name=worker --task_index=0 --issync=1#python sync_dist_train.py --job_na...原创 2019-02-25 16:55:39 · 3714 阅读 · 1 评论 -
TensorFlow实现半同步训练
一、半同步例子:#coding=utf-8#python sync_dist_train.py --job_name=ps --task_index=0 --issync=1#python sync_dist_train.py --job_name=worker --task_index=0 --issync=1#python sync_dist_train.py --job_nam...原创 2019-02-25 17:38:21 · 1236 阅读 · 2 评论