希望支持AMD RoCm加速

[rikimeow]

希望支持AMD RoCm加速🙏

补充信息：
我的GPU是[AMD Ryzen™ Al Max+ 395]
(https://www.amd.com/en/products/processors/laptop/ryzen/ai-300-series/amd-ryzen-ai-max-plus-395.html)
gpu_type=gfx1151
rocm版本：6.4.1

[myhloli]

自己装个支持rocm的torch就可以了吧，我们没有设备没法测试这个，可以自行摸索下

[rikimeow]

好的🙏我试试

[rikimeow]

我用Rocm/therock社区的一个pytorch容器，在GPU上跑起来了。
https://github.com/ROCm/TheRock/pkgs/container/therock_pytorch_dev_ubuntu_24_04_gfx1151/442724611?tag=main

[rikimeow]

pytorch官方2.8.0版本预计支持gfx1151，7/7/2025发布。
pytorch/pytorch#156745 (comment)

[rikimeow]

补充：由于gfx1151和gfx1100兼容，可以临时修改变量模拟成gfx1100， export HSA_OVERRIDE_GFX_VERSION=11.0.0
再安装官方的nigtly预览版2.8.0（pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4），就可以跑起来了