机器视觉-机器视觉-尝试Windows下使用Docker 容器进行AMD ROCm加速(3)

结论

虽然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