在使用TensorFlow深度学习的时候,会默认占用最大显存,如何设置显存占用率,总结如下: I. 两种方法查看显存使用情况 II. 设置TensorFlow显存占用率 III. 代码插入位置 IV. 使用对比 V. 参考资料 I. 两种方法查看显存使用情况 方法1:CMD窗口中输入命令nvidia-smi 方法2: 使用软件AIDA64(推荐),实时监测电脑硬件使用情况 电脑闲置可用显存为7443 MB。 II. 设置TensorFlow显存占用率 方法1:按需动态分配显存(更灵活) config = tf.ConfigProto() config.gpu_options.allow_growth = True