[881]内存不足RuntimeError: CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0； 2.00 GiB total cap...

原因一：找到错误点，增加以下语句： with torch.no_grad(): outputs = Net_(inputs) —错误代码的位置。 原因二：GPU没有选对 os.environ[“CUDA_VISIBLE_DEVICES”] = “0, 2, 3” 查看性能，发现nVidia的只有GPU1，所以改成如下： os.environ[“CUDA_VISIBLE_DEVICES”] = “0

>>阅读原文<<