运行代码时出现的错误如下:
RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB (GPU 0; 4.00 GiB total capacity; 2.91 GiB already allocated; 166.40 KiB free; 2.93 GiB reserved in total by PyTorch)
看了一下自己的GPU
肯定是支持代码运行的,于是看了一下自己的代码:
查找了其他的博文,先试了最简单的将batch_size改小了点,我改成了2,运行就成功了
现在还在运行中……