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) 打赏
软件无法使用？点击查看常见问题说明>>
分享到

相关推荐

最近更新

优云智算HappyHorse AI 视频生成API KEY WebUI-诺瓦小站

优云智算HappyHorse AI 视频生成API KEY WebUI

大家可能平常刷视频的时候已经发现了，进入2026年AI短剧极其火爆，不管是短视频平台还是拼多多等购物平台，短剧视频里经常能刷到AI漫剧。前几年短剧公司拍真人短剧赚的盆满钵满，进入2026年短剧公司被AI打的毫无还手之力。借助AI的快速发展，...

赞(0)AI视频
LTX-2.3 视频生成免安装部署整合包软件下载-诺瓦小站

LTX-2.3 视频生成免安装部署整合包软件下载

LTX-2 是由 Lightricks 开发的首个基于 DiT（扩散变换器）架构的音视频基础模型，能够在一个统一模型中同时生成高质量的视频与同步音频。与以往需要分别处理视频和音频的方案不同，LTX-2 将两者深度融合，实现真正的音画同步生成...

赞(1)AI视频

PDF转word软件FreeP2W免安装版下载

这软件还是很早的时候一个用户让做的，发现没分享过，现在分享一下。 软件主要功能就是把PDF文档转换为word文档 软件使用很简单，把需要处理的pdf文档复制到input文件夹内，双击启动软件.exe，等待处理完成即可。 注意事项 软件只支持...

赞(0)AI文本
FunASR语音识别转文字软件区分说话人版-诺瓦小站

FunASR语音识别转文字软件区分说话人版

FunASR是一款优秀的基础语音识别框架，配合阿里通义团队开发的其它语音识别模型，能够实现非常好语音转文字效果，尤其在中文语音识别方面，鉴于有些用户需要对语音识别内容区分说话人，我重新做了一版区分说话人的版本。软件支持多语言识别，但是还是尽...

赞(0)AI音频
答疑咨询/软件定制修改/AI进阶玩法

热门文章

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

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

支付宝扫一扫

微信扫一扫