我的GPU版本是NVIDIA GeForce MX150,执行VoxelMorph代码时显示Resource exhausted: OOM when allocating tensor with shape[1,16,160,192,224] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
刚开始以为是batch_size太大,后来按照https://blog.csdn.net/weixin_41770169/article/details/80349088查看GPU占用率,发现GPU并没有被使用,GPU memory usage处显示N/A,即not available。
1.首先按照上述链接说明,使用如下命令
>>> import tensorflow as tf
>>> sess = tf.Session