Jupyter notebook 解决程序运行持续占用内存问题

在jupyter中运行程序后,点击kernel中的shutdown。如下图所示

或者

当然,在一开始的时候加上如下代码,进行显存管理是必要的:

import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0'   #指定第一块GPU可用
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5  # 程序最多只能占用指定gpu50%的显存
config.gpu_options.allow_growth = True      #程序按需申请内存
sess = tf.Session(config = config)

 


版权声明:本文为weixin_41126303原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。