0
点赞
收藏
分享

微信扫一扫

机器视觉-机器视觉-尝试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_v8https://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://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


举报

相关推荐

0 条评论