大大你好,我上一个回复已经试过将TORCH_HOME改为模型的绝对路径,但不幸的是,依然报错;另外我本人pc上没有mlu270,因此程序无法找到卡设备,以上步骤是通过cpu进行推理的
您好,从第二个截图中看到,state_dict=torch.load(TORCH_HOME,map_location='cpu') 。为了明确的话,可以直接将TORCH_HOME修改为模型的绝对路径
2.将TORCH_HOME更改为模型(pth文件)路径,但依然报错,错误是(NotADirectory:xxx/xx.pth/int8/),似乎依然在寻找int8这个文件夹
您好,从图中看有以下几个问题。(1)“NO MLU DEVICE" ,说明程序没有找到卡设备。可能是docker启动脚本中,没有把卡映射进来。(2) TORCH_HOME中存放的是模型路径,由于docker中没有提供模型,所以可以设置为已有模型的路径。
请登录后评论