python不结束进程清除tensorflow占用显存
时间: 2023-09-02 21:05:50 浏览: 136
解决TensorFlow训练内存不断增长,进程被杀死问题
可以使用以下代码来清除 TensorFlow 占用的显存:
```python
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` 库。
阅读全文