python跑完清空内存_python – 在模型执行后清除Tensorflow GPU内存

2016年6月(

https://github.com/tensorflow/tensorflow/issues/1727)的git问题表明存在以下问题：

currently the Allocator in the GPUDevice belongs to the ProcessState,

which is essentially a global singleton. The first session using GPU

initializes it, and frees itself when the process shuts down.

因此,唯一的解决方法是使用进程并在计算后关闭它们.

示例代码：

import tensorflow as tf

import multiprocessing

import numpy as np

def run_tensorflow():

n_input = 10000

n_classes = 1000

# Create model

def multilayer_perceptron(x, weight):

# Hidden layer with RELU activation

layer_1 = tf.matmul(x, weight)

return layer_1

# Store layers weight & bias

weights = tf.Variable(tf.random_normal([n_input, n_classes]))

x = tf.placeholder("float", [None, n_input])

y = tf.placeholder("