##起因
每次在服务器跑深度学习代码,我的数据量很小。但是内存直接占满。如下图
可以设置内存占用情况。
解决办法
### 在前三行加上代码:
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)])
结果
解决内存占用问题