[人工智能] “win10下pytorch1.9.0+cuda11.1 gpu显存充足但pytorch可分配内存不足”问题解决

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 人工智能 -> “win10下pytorch1.9.0+cuda11.1 gpu显存充足但pytorch可分配内存不足”问题解决 -> 正文阅读

[人工智能]“win10下pytorch1.9.0+cuda11.1 gpu显存充足但pytorch可分配内存不足”问题解决

Traceback (most recent call last):
  File "C:/Users/CAF_d201/PycharmProjects/pythonProject/代码资源_PyTorch深度学习入门/CH4/TransferLearning.py", line 125, in <module>
    train(alexnet,criterion,optimizer,epochs=2)
  File "C:/Users/CAF_d201/PycharmProjects/pythonProject/代码资源_PyTorch深度学习入门/CH4/TransferLearning.py", line 89, in train
    loss.backward()
  File "D:\Anaconda3_5\envs\pytorch\lib\site-packages\torch\_tensor.py", line 255, in backward
    torch.autograd.backward(self, gradient, retain_graph, create_graph, inputs=inputs)
  File "D:\Anaconda3_5\envs\pytorch\lib\site-packages\torch\autograd\__init__.py", line 149, in backward
    allow_unreachable=True, accumulate_grad=True)  # allow_unreachable flag
RuntimeError: CUDA out of memory. Tried to allocate 144.00 MiB (GPU 0; 2.00 GiB total capacity; 282.16 MiB already allocated; 118.23 MiB free; 294.00 MiB reserved in total by PyTorch)