环境信息
项目 信息
GPU 8 × K100_AI (AMD)
PyTorch 2.5.1
ROCm/HIP HIP 5.x
CUDA可见 torch.cuda.is_available() = True
在 AMD K100_AI GPU 上运行 PyTorch 模型时,出现 VMFault 错误,进程崩溃:
Could not open /var/log/hylog/.
Invalid address access: 0x149d18402000, Error code: 3.
>>>>>>>> KERNEL VMFault !!!! <<<<<<
>>>>>>>> PID: 2005192, SIGNAL: 0 !!!! <<<<<<
>>>>>>>>
>>>>>>>
>>>>>>
>>>>>
>>>>
>>>
>>
GPU 可被 PyTorch 识别:
torch.cuda.is_available() = True
torch.cuda.device_count() = 8
小张量操作正常:
x = torch.randn(1000, 1000).cuda() # ✅ 成功
模型加载到 GPU 成功:
model = model.cuda() # ✅ 成功
前向传播时触发 VMFault:
output = model(images, mode='sample') # ❌ VMFault 崩溃

各位好心的大佬!!救救孩子吧!!为啥啊!!