今天试了一下,缩小版的数据集,依旧无法运行。
报错为RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 6.00 GiB total capacity; 4.58 GiB already allocated; 6.94 MiB free; 4.65 GiB reserved in total by PyTorch)
这证明是有什么软件提前占用了gpu。
Windows下查看GPU(NVIDIA)使用情况_墨理三生-CSDN博客
分别输入cmd
C:\Program Files\NVIDIA Corporation\NVSMI
nvidia-smi.exe
相关错误总结
Pytorch运行错误:CUDA out of memory处理过程 - 灰信网(软件开发博客聚合)