太长不看版： windows环境下无法通过pytorch使用A卡的gpu加速。

A卡跑DL的事情比较复杂，咱们一层一层来说这个事情。

gpu加速计算方面，AMD因为起步晚，前几年A家的cpu差点儿被I家干死，没有精力分神搞gpu，好不容易cpu这两年翻身了，回过头来N卡的cuda早已成为事实标准，这中间至少10年的差距，AMD再造一套标准已经不靠谱了，所以A家在DL方面的一个策略是抄袭cuda，实现api兼容，用这种方式把tensorflow/pytorch这些已经对cuda做过适配的DL框架整体搬到自家显卡上。

AMD的这个东西叫做ROCm - Radeon Open Compute Platform，Radeon锐龙就是A家显卡的品牌。

截至2021-12-14，ROCm发布的最新版本为4.5.2。预计2022年第一季度发布ROCm-5.0.0（这个是猜的，ROCm没有官方路线图，跳票非常厉害）。

那么ROCm的兼容性如何呢？

首先只支持linux（no windows, no macos），然后只支持有数的几个linux发行版（centos, ubuntu），最后对内核也有要求，ubuntu-20.04.4临近发布，据说要升级到kernel-5.13，社区已经报出amdpro无法安装，需要手工退回到kernel-5.11。

https://github.com/RadeonOpenCompute/ROCm/issues/1658

是不是安装了指定的linux发行版，再对应好指定的kernel就能跑pytorch的gpu加速了呢？还不行。

因为ROCm支持的gpu显卡极其有限，理论上只支持gfx900(vega56, vega64), gfx906(radeon vii), gfx908(m100)。

为啥说理论上？

那么windows的希望在哪里？