python清除tensorflow占用显存

在 Python 中，你可以使用以下代码清除 TensorFlow 占用的显存：

import tensorflow as tf

tf.reset_default_graph()
sess = tf.compat.v1.Session()

# 以下是你的 TensorFlow 代码

# 清除 TensorFlow 占用的显存
tf.compat.v1.reset_default_graph()
sess.close()

在这个例子中，`tf.reset_default_graph()` 用于清除默认图形。`tf.compat.v1.Session()` 创建一个新的 TensorFlow 会话。在运行完 TensorFlow 代码后，`tf.compat.v1.reset_default_graph()` 用于清除默认图形，`sess.close()` 用于关闭 TensorFlow 会话并释放显存。

相关问题

python不结束进程清除tensorflow占用显存

可以使用以下代码来清除 TensorFlow 占用的显存：

import tensorflow as tf
from numba import cuda

# 创建 TensorFlow Session 对象
sess = tf.Session()

# ...

# 关闭 TensorFlow Session 对象
sess.close()

# 清除 TensorFlow 占用的显存
cuda.select_device(0)
cuda.close()

其中，`cuda.select_device(0)` 表示选择第 0 个 GPU 设备，`cuda.close()` 表示清除该设备占用的显存。如果有多个 GPU 设备，可以选择不同的设备进行清除。此外，还需要安装 `numba` 库。

清除tensorflow占用显存

在使用 TensorFlow 进行机器学习训练时，可能会遇到显存不足的问题，这时候可以使用以下方法来清除 TensorFlow 占用的显存：

1. 在训练代码中加上以下代码：

import tensorflow as tf
from keras.backend.tensorflow_backend import set_session

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
set_session(sess)

这段代码会让 TensorFlow 动态申请显存，避免一次性申请过多导致显存不足。

2. 如果还是出现显存不足的情况，可以使用以下命令来清除 TensorFlow 占用的显存：

sudo lsof /dev/nvidia* | grep python | awk '{print $2}' | xargs sudo kill -9

这条命令会杀掉所有正在占用 NVIDIA 显卡的 Python 进程，包括 TensorFlow。注意，这会导致所有正在运行的 TensorFlow 程序停止，因此请谨慎使用。