AppleM1PyTorchmps内存泄漏?
创始人
2024-09-10 19:00:52
0

近期有很多人报告在 Apple M1 芯片上使用 PyTorch 和 mps 时出现内存泄漏的情况。根据这个问题的解决方案,可以通过将 PyTorch 降级到 1.7.1 或更低版本,使用 OpenBLAS 代替 Accelerate Framework,以及关闭 mps 来解决这个问题。以下是示例代码,用于降级 PyTorch 和使用 OpenBLAS:

pip uninstall torch
export NO_CUDA=1
pip install torch==1.7.1+cpu torchvision==0.8.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
pip install cython
pip uninstall numpy
pip install numpy==1.19.3

接下来,需要安装 OpenBLAS,可以使用 Homebrew 命令来安装:

brew install openblas

最后,可以通过设置环境变量关闭 mps:

export TORCH_CUDA_IPC_DISABLE=1

这样,在 Apple M1 上使用 PyTorch 和 mps 时就不会出现内存泄漏问题了。

相关内容

热门资讯

实测揭晓!微扑克职业代打,微扑... 实测揭晓!微扑克职业代打,微扑克有挂么(微扑克)切实存在有挂(有挂解惑)-哔哩哔哩;1、许多玩家不知...
研究成果!微扑克辅助软件,微扑... 研究成果!微扑克辅助软件,微扑克有挂么(微扑克)本来存在有挂(有挂实锤)-哔哩哔哩微扑克辅助软件软件...
终于清楚!微扑克辅助挂,微扑克... 终于清楚!微扑克辅助挂,微扑克如何让系统发好牌(微扑克)确实是有挂(有挂秘籍)-哔哩哔哩1、微扑克如...
重大通报!微扑克ai辅助工具,... 重大通报!微扑克ai辅助工具,微扑克透牌(微扑克)切实有挂(有挂猫腻)-哔哩哔哩进入游戏-大厅左侧-...
普及知识!微扑克有ai吗,微扑... 普及知识!微扑克有ai吗,微扑克如何让系统发好牌(微扑克)本来有挂(存在有挂)-哔哩哔哩1、上手简单...
一分钟教会你!微扑克有挂么,微... 一分钟教会你!微扑克有挂么,微扑克ai机器人(微扑克)好像是真的有挂(真的有挂)-哔哩哔哩1、微扑克...
玩家实测!微扑克辅助工具,微扑... 玩家实测!微扑克辅助工具,微扑克如何让系统发好牌(微扑克)真是存在有挂(详细教程)-哔哩哔哩1、全新...
玩家必看科普!微扑克辅助提高胜... 玩家必看科普!微扑克辅助提高胜率,微扑克透牌(微扑克)原来有挂(有挂神器)-哔哩哔哩微扑克辅助提高胜...
查到实测辅助!微扑克辅助工具怎... 查到实测辅助!微扑克辅助工具怎么下载,微扑克有挂么(微扑克)确实有挂(有挂猫腻)-哔哩哔哩;1、微扑...
最新研发!微扑克有挂么,微扑克... 最新研发!微扑克有挂么,微扑克有挂么(微扑克)果然存在有挂(真的有挂)-哔哩哔哩1)微扑克有挂么辅助...