##起因

每次在服务器跑深度学习代码，我的数据量很小。但是内存直接占满。如下图

可以设置内存占用情况。

解决办法

### 在前三行加上代码：

import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
tf.config.experimental.set_virtual_device_configuration(gpus[0],[tf.config.experimental.VirtualDeviceConfiguration(memory_limit=2048)])

结果

 解决内存占用问题