
Tensorflow
Microstrong0305
菜鸟程序猿
展开
-
TensorFlow中使用GPU方法
1. Tensorflow中指定程序在哪一块GPU上训练Python中代码:import os# 使用第一张与第三张GPU卡os.environ["CUDA_VISIBLE_DEVICES"] = "0, 2"命名行代码:CUDA_VISIBLE_DEVICES=0,2 python train.py 2. 按需增加GPU的内存import tensorflow ...原创 2019-03-25 15:48:23 · 6817 阅读 · 1 评论 -
深度学习中Tensorflow多显卡多人使用问题集
微信公众号1.多显卡环境在某块显卡被占用时在其他显卡运行返回:Segmentation fault (core dumped)当我在用编号为1的显卡时,其他人使用其他显卡时就会报这个错。而其他人使用编号0显卡时,我使用编号2显卡也会报错。在网上也可以看到此类问题。虽然tensorflow可以指定显卡(参见:Using GPUs),如果前端用keras时也就无从指定。其实有一劳永...原创 2019-03-04 11:27:51 · 1971 阅读 · 0 评论 -
Tensorflow中scope命名方法
两篇文章掌握Tensorflow中scope用法:【1】Tensorflow中scope命名方法(本文)【2】Tensorflow中tf.name_scope() 和 tf.variable_scope() 的区别微信公众号1. tf.name_scope()在 Tensorflow 当中有两种途径生成变量 variable,一种是 tf.get_variable(),另一种...原创 2018-12-04 17:02:27 · 2363 阅读 · 0 评论