AI软件下载
有趣网站推荐及实用软件下载
当前位置：诺瓦小站 问题记录 python 正文

torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 2.88 GiB. GPU 0 has a total capacity of 44.40 GiB of which 1.59 GiB is free. Including non-PyTorch memory,

2025-10-31 分类：python 评论(0)

Python应用报错：

torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 2.88 GiB. GPU 0 has a total capacity of 44.40 GiB of which 1.59 GiB is free. Including non-PyTorch memory, this process has 42.80 GiB memory in use. Of the allocated memory 38.56 GiB is allocated by PyTorch, and 3.75 GiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)

出错原因就是显存不足，报错已经提示解决方法了，设置环境变量减少内存碎片化，在Python脚本里添加如下代码：

import os
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'


AI软件用不了？2元爽玩4090： 立即体验>>

热门大语言模型API免费体验： 立即获取>>

软件催更及1对1人工答疑支持： https://nuowa.net/1806
赞(0) 打赏
软件无法使用？点击查看常见问题说明>>
分享到

相关推荐

最近更新

最新局域网手机电脑文件互传工具推荐

最近每天都要手机向电脑传多个文件，又需要用到手机电脑互传工具。微信有这个功能，但是我不怎么在电脑上登微信，传文件的话还要打开微信进行登录，有些麻烦。而且微信传文件太慢了，所以不想用微信传文件。 又想到了以前用的一个专用的局域网手机电脑文件互...

赞(0)实用软件
Claude 再放大招，Sonnet 5 来了-诺瓦小站

Claude 再放大招，Sonnet 5 来了

一觉醒来Anthropic又放大招了，Claude Sonnet 5来了。 不得不说国外大语言模型更新的是真快的，各家LLM版本更新迭代频率都很高，Claude 刚发布Opus 4.8 不久，这又发布了Sonnet 5.Anthropic ...

赞(0)趣站
祝贺凡人修仙传2026年新年番开播同时在线人数超64万-诺瓦小站

祝贺凡人修仙传2026年新年番开播同时在线人数超64万

今天6月13日凡人修仙传新年番开播，同时在线人数破64万（非最高在线人数，只是我看到的在线人数），作为凡人5年老粉，必须发个帖祝贺一下。 我比较喜欢看动漫，各种类型看了很多，具体不清楚多少，像斗罗斗破之类看了一大半终究是没能看下去，还是雾山...

赞(2)趣站

热门文章

觉得文章对你有帮助就打赏一下作者

非常感谢你的打赏，我将有更多的动力继续提供优质内容，让我们一起创建更加美好的世界！

支付宝扫一扫

微信扫一扫