Pytorch报错RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available
·
在“从零开始大模型开发与微调”的学习过程中，在第十章predicate.py执行时，报如标题的错误，原因为我的环境为apple M1处理器运行Pytorch，不是cuda，而是MPS显卡，针对predicate.py的两处代码需要修改：
1）device = "cuda" （原始代码）
if(torch.backends.mps.is_available()): device = "mps"（在上面原始代码的下面追加此行代码）
2）model.load_state_dict(torch.load("./saver/model.pth"))
修改为：model.load_state_dict(torch.load("./saver/model.pth", map_location=device))
保存运行，完美！
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
4
0- 0
已为社区贡献2条内容
所有评论(0)