Tensorflow使用GPU运行

参考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, ...)