tensorflow学习
px_cmd
这个作者很懒,什么都没留下…
展开
-
tensorflow学习
1、tensorflow:模型的保存与恢复参考:tensorflow 1.0 学习:模型的保存与恢复(Saver)2、指定GPU设备号和个数方法一:如果你在终端指定GPU个数和ID号,如果电脑有多个GPU,tensorflow默认全部使用。如果想只使用部分GPU,可以设置CUDA_VISIBLE_DEVICES。在调用python程序时,可以使用:CUDA_VISIBLE_DEVI...原创 2019-03-09 15:25:07 · 126 阅读 · 0 评论 -
tensorflow学习(1)------- tf.ConfigProto()配置Session运行参数&&GPU设备指定
tf.ConfigProto()函数用在创建session的时候,用来对session进行参数配置:config = tf.ConfigProto(allow_soft_placement=True, allow_soft_placement=True)config.gpu_options.per_process_gpu_memory_fraction = 0.4 #占用40%显存s...原创 2019-03-15 16:18:56 · 253 阅读 · 0 评论 -
tensorflow学习(2)tf.InteractiveSession()和tf.Session()区别
tf.InteractiveSession():它能让你在运行图的时候,插入一些计算图,这些计算图是由某些操作(operations)构成的。这对于工作在交互式环境中的人们来说非常便利,比如使用IPython。tf.Session():需要在启动session之前构建整个计算图,然后启动该计算图。意思就是在我们使用tf.InteractiveSession()来构建会话的时候,我们可以先构建...原创 2019-03-22 10:17:06 · 649 阅读 · 0 评论