问题:服务器上多块卡,使用其中一张训练的模型,在本地预测的时候报错。
解决:在torch.load中加入map_location,指定一块卡
标签:count,object,Attempting,RuntimeError,torch,CUDA,device From: https://www.cnblogs.com/Liang-ml/p/17767331.html
问题:服务器上多块卡,使用其中一张训练的模型,在本地预测的时候报错。
解决:在torch.load中加入map_location,指定一块卡
标签:count,object,Attempting,RuntimeError,torch,CUDA,device From: https://www.cnblogs.com/Liang-ml/p/17767331.html