luckynote

在Tensorflow 训练模型时报错提示：failed to allocate 3.77G (4046333952 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory虽然会报出显存溢出问题，但不影响正常训练，不过笔者还是想知道这个问题是怎么来的。废话不多说，先上session初始化的代码gpu_optio...

问题内容如下 InvalidArgumentError (see above for traceback): Cannot assign a device to node 'train/final/fc3/b/Momentum': Could not satisfy explicit device specification '/device:GPU:0' because no devices m

笔者在ubuntu上跑Tensorflow的程序的时候，中途使用了Win+C键结束了程序的进行，但是GPU的显存却显示没有释放，一直处于被占用状态。 使用命令 watch -n 1 nvidia-smi 显示如下 两个GPU程序都在执行中，实际上GPU:0已经被笔者停止了，但是GPU没有释放，进程还在继续，所以只有采用暴力手段了，将进程手动关闭掉，进程编号如图中红线部分，由于笔者在两个G

问题详情： Caused by op u’save/RestoreV2’, defined at: File “demo.py”, line 25, in result_dict = news_demo.newsAggreg({image_path}) File “/home/rszj/liutao/news_aggreg/news_demo.py”, line 32,

最近笔者在做GPU显存资源分配的研究，发现在tf中gpu一些实用的方法和接口，共享出来，供大家参考学习，如有问题，欢迎留言讨论。

tf.set_random_seed(seed) 用于：设置图级随机seed。seed操作有两种设置方式：图级seed 和 操作级seed。并且两种seed是关联使用的。相互作用如下：1.如果没有设置图形级别和操作seed，则使用随机seed进行操作。2.如果设置了图级seed，但操作seed没有设置：系统确定性地选择与图级seed一起的操作seed，以便获得唯一的随机序列。...

采用python map 的正确写法如下mat_ratio = list(map(lambda x: 1.0/x if not x == 0.0 else float(64.0), mat_ratio))但是，在有的时候，我们需要x是一个tensor，mat_ratio定义的是一个op graph，按照常规思维，进行如下改写：mat_ratio = tf.map_fn(lamb...