結論

雖然WSL下無法使用ROCm加速, 但宿主機是Linux下是可以的, 而且核顯也是可以使用ROCm加速的.

ROCm

ROCm 是AMD開發的高性能計算平臺, 對應的是CUDA. 使用ROCm后AMD的GPU也可以實現高性能數學計算. ROCm支持的平臺有:

• 支持Linux
• 支持Docker
• 不支持Windows, 也不支持WLS和虛擬機模式

RX5000/6000系列顯卡: export HSA_OVERRIDE_GFX_VERSION=10.3.0, 如果不行的話設置 export HSA_OVERRIDE_GFX_VERSION=9.0.0
RX7000系列及以下顯卡: export HSA_OVERRIDE_GFX_VERSION=11.0.0
RX580 環境變量: export HSA_OVERRIDE_GFX_VERSION=8.0.3

核顯使用使用ROCm加速

https://github.com/harakas/amd_igpu_yolo_v8
https://www.gabriel.urdhr.fr/2022/08/28/trying-to-run-stable-diffusion-on-amd-ryzen-5-5600g/

WSL 下的問題

https://unix.stackexchange.com/questions/715847/wsl2-issue-installing-new-kernel/715922#715922

更多文章

https://zhuanlan.zhihu.com/p/566112395
https://zhuanlan.zhihu.com/p/656480759
https://blog.csdn.net/qq_42183962/article/details/132574429
https://bbs.oldmantvg.net/thread-26091.htm
https://rocm.docs.amd.com/projects/install-on-linux/en/develop/how-to/3rd-party/pytorch-install.html#using-docker-with-pytorch-pre-installed