美文网首页Tensorflow使用记录
Tensorflow使用GPU运行

Tensorflow使用GPU运行

作者: 吐舌小狗 | 来源:发表于2018-01-14 20:23 被阅读254次

参考https://tensorflow.google.cn/tutorials/using_gpu

1.根据程序的需要动态分配GPU

本人常用的也是最佳的使用方法

# -*- coding: utf-8 -*-

import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)

2.设置固定的值

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4
session = tf.Session(config=config, ...)

相关文章

网友评论

    本文标题:Tensorflow使用GPU运行

    本文链接:https://www.haomeiwen.com/subject/fxdloxtx.html