前言上一篇文章里我开始折腾在 Windows11 上编译 ROCm 版的 PyTorch虽然折腾失败了但积累了一些经验。这不第二天就编译成功了。编译产物爽啊/* by yours.tools - online tools website : yours.tools/zh/calctime.html */ -a--- 2026/1/5 15:58 151148776 torch-2.9.1rocm7.11.0a20260104-cp312-cp312-win_amd64.whl -a--- 2026/1/5 15:59 514989 torchaudio-2.9.0rocm7.11.0a20260104-cp312-cp312-win_amd64.whl -a--- 2026/1/5 16:01 1247290 torchvision-0.24.0rocm7.11.0a20260104-cp312-cp312-win_amd64.whlWindows AMD ROCm PyTorch懂不懂这些词凑在一起的含金量啊要点总结下来这次编译功能的要点就俩。使用官方版的 ROCm之前因为网上查到的资料都说官方还没推出 gfx103x 的 ROCm所以我安装了一个第三方预构建好的版本可能是太老了还是咋的编译 PyTorch 时老是失败。这次我使用了这个命令安装/* by yours.tools - online tools website : yours.tools/zh/calctime.html */ pip install --index-url https://rocm.nightlies.amd.com/v2-staging/gfx103X-dgpu/ --pre rocm[libraries,devel]你可以在这些位置尝试找到对应的 ROCm 软件包https://rocm.nightlies.amd.com/v2-staging/https://rocm.nightlies.amd.com/v2/https://rocm.devreleases.amd.com/v2https://rocm.devreleases.amd.com/v2-staging别迷信第三方包能用官方的就用官方的没有就自己编译 ROCm很幸运我这个卡有官方的 ROCm 包。使用旧版本 PyTorch这一点上篇文章也说了最新版可能不行我在网上看有人编译 2.10 版本成功了。其实之前折腾的时候已经隐约感觉到了PyTorch 版本太新在 Windows ROCm 这条路径上更容易踩坑。网上已经有人确认2.10有人成功2.11存在各种不确定性这次我直接选了更保守的方案PyTorch 2.9.1事实证明这个选择是对的。至于 2.11 行不行以后有精力再继续折腾现在先享受成果。安装 wheel既然编译好了那就可以安装试用一下。按顺序安装这三个文件注意torch 必须最先安装pip install .\torch-2.9.1rocm7.11.0a20260104-cp312-cp312-win_amd64.whl pip install .\torchvision-0.24.0rocm7.11.0a20260104-cp312-cp312-win_amd64.whl pip install .\torchaudio-2.9.0rocm7.11.0a20260104-cp312-cp312-win_amd64.whlWindows AMD 显卡炼丹通道已打通。下一步做啥显卡加速能用了那能做的事情就很多了。深度学习跑起来AI画图玩起来~下一步我想先试试一些简单的深度学习模型训练任务把 AI 画图重新玩起来重点试试 Z-Image : https://huggingface.co/Comfy-Org/z_image_turbo这个系列后续还有很多值得记录分享的欢迎继续关注。微信公众号「程序设计实验室」 专注于互联网热门新技术探索与团队敏捷开发实践包括架构设计、机器学习与数据分析算法、移动端开发、Linux、Web前后端开发等欢迎一起探讨技术分享学习实践经验。