AMDROCm和Pytorch在Navi10（RX5700XT）上使用HSA_OVERRIDE_GFX_VERSION=10.3.0时无法运行
创始人
2024-08-09 15:30:18
0次
首先，建议更新ROCm和amdgpu-pro驱动程序。如果问题仍然存在，请尝试使用以下命令：
export HSA_OVERRIDE_GFX_VERSION=1010 # or 1011/1012
export HIP_VISIBLE_DEVICES=0 # (0/1/2/3 etc corresponding to gpu ids)
这将在环境变量中设置正确的GFX版本，并将可见设备设置为您要使用的GPU。然后，在Pytorch代码中，确保在初始化GPU时使用hip包而不是torch。例如：
import torch
# Replace this line
device = torch.device('cuda')
# with this
device = torch.device('hip')
# The rest of the code should work as expected
这应该解决问题并让您顺利运行您的Pytorch代码。
上一篇：amdrezenubuntu
下一篇：amd深度学习主机
相关内容
热门资讯
安卓换鸿蒙系统会卡吗,体验流畅...
最近手机圈可是热闹非凡呢！不少安卓用户都在议论纷纷，说鸿蒙系统要来啦！那么，安卓手机换上鸿蒙系统后，...
安卓系统拦截短信在哪,安卓系统...
你是不是也遇到了这种情况：手机里突然冒出了很多垃圾短信，烦不胜烦？别急，今天就来教你怎么在安卓系统里...
app安卓系统登录不了,解锁登...
最近是不是你也遇到了这样的烦恼：手机里那个心爱的APP，突然就登录不上了？别急，让我来帮你一步步排查...
安卓系统要维护多久,安卓系统维...
你有没有想过，你的安卓手机里那个陪伴你度过了无数日夜的安卓系统，它究竟要陪伴你多久呢？这个问题，估计...
windows官网系统多少钱
Windows官网系统价格一览：了解正版Windows的购买成本Windows 11官方价格解析微软...
安卓系统如何卸载app,轻松掌...
手机里的App越来越多，是不是感觉内存不够用了？别急，今天就来教你怎么轻松卸载安卓系统里的App，让...
怎么复制照片安卓系统,操作步骤...
亲爱的手机控们，是不是有时候想把自己的手机照片分享给朋友，或者备份到电脑上呢？别急，今天就来教你怎么...
安卓系统应用怎么重装,安卓应用...
手机里的安卓应用突然罢工了，是不是让你头疼不已？别急，今天就来手把手教你如何重装安卓系统应用，让你的...
iwatch怎么连接安卓系统,...
你有没有想过，那款时尚又实用的iWatch，竟然只能和iPhone好上好？别急，今天就来给你揭秘，怎...
iphone系统与安卓系统更新...
最近是不是你也遇到了这样的烦恼？手机更新系统总是失败，急得你团团转。别急，今天就来给你揭秘为什么iP...