Navi10(RX 5700 XT)搭配ROCm+PyTorch时HSA_OVERRIDE_GFX_VERSION=10.3.0失效求助
关于RX 5700 XT通过ROCm运行PyTorch的问题
我在Stack Overflow上看到帖子《AMD ROCm with Pytorch on Navi10 (RX 5700 / RX 5700 XT)》,里面推荐使用HSA_OVERRIDE_GFX_VERSION=10.3.0环境变量来让RX 5700 XT显卡通过ROCm运行PyTorch,但尝试后未能成功。
操作步骤
sudo pacman -S python-pytorch-opt-rocm git clone https://github.com/pytorch/examples.git cd examples/mnist HSA_OVERRIDE_GFX_VERSION=10.3.0 python3 main.py
运行结果
执行MNIST训练时,GPU仅发热但无任何训练进度,输出内容提示ROCm设备初始化阶段存在异常,无法正确调度GPU内核执行任务。
相关环境信息
- rocminfo输出显示:Navi10(RX 5700 XT)的原生GFX版本为10.1.0,ROCm基础组件已加载,但设备兼容性检测存在不匹配提示。
- lscpu输出确认我的CPU支持AVX2指令集。
额外尝试
我也试过安装python-pytorch-rocm包,不过考虑到CPU支持AVX2,python-pytorch-opt-rocm包应该是更合适的选择。
请问针对Navi10 GPU的这个workaround目前是否仍有效?或是我在配置过程中遗漏了什么关键步骤?
内容的提问来源于stack exchange,提问作者Eric Armbruster
相关产品推荐
相关产品推荐

