赞
踩
在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)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。