Pytorch载入保存的模型更换gpu卡号如何设置map_location
Pytorch在保存模型时，会把训练过程中使用的设备号（GPU:0, CPU等）也一起保存下来，当load保存的模型时，会默认把权重加载到训练时使用的设备上，要修改加载到的卡号可以如下：
torch.load('your_model.pth', map_location=lambda storage, loc : storage.cuda(1))
Pytorch在保存模型时，会把训练过程中使用的设备号（GPU:0, CPU等）也一起保存下来，当load保存的模型时，会默认把权重加载到训练时使用的设备上，要修改加载到的卡号可以如下：
torch.load('your_model.pth', map_location=lambda storage, loc : storage.cuda(1))
【推荐】国内首个AI IDE，深度理解中文开发场景，立即下载体验Trae
【推荐】编程新体验，更懂你的AI，立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包，你的智能百科全书，全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell：AI 加持，快人一步
· 25岁的心里话
· 闲置电脑爆改个人服务器（超详细） #公网映射 #Vmware虚拟网络编辑器
· 零经验选手，Compose 一天开发一款小游戏！
· 因为Apifox不支持离线，我果断选择了Apipost！
· 通过 API 将Deepseek响应流式内容输出到前端