问题遇到的现象和发生背景 在训练代码时遇到
InternalError: Failed copying input tensor from /job:localhost/replica:0/task:0/device:CPU:0 to /job:localhost/replica:0/task:0/device:GPU:0 in order to run _EagerConst: Dst tensor is not initialized.
问题相关代码,请勿粘贴截图
显示内存不足
但是NVIDIA-smi显示还有显存