python清除tensorflow占用显存
时间: 2023-08-28 07:04:32 浏览: 89
解决Tensorflow占用GPU显存问题
在 Python 中,你可以使用以下代码清除 TensorFlow 占用的显存:
```python
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 会话并释放显存。
阅读全文