运行这段程序的时候出现如下错误:
RuntimeError: CUDA out of memory. Tried to allocate 1.38 GiB (GPU 0; 7.80
京公网安备 11010502049817号