- 博客(2)
- 收藏
- 关注
RSS订阅原创 模型在一张显卡上训练完后，在另一张显卡上测试，模型失效
my_model.state_dict()的device是cuda1，在torch.load把cuda1的参数赋给cuda2上的模型时，参数的值全为零。（不是到是不是torch版本的问题）解决：先将cuda1的数据移动到cpu上，再移动到cuda2上，x与x2的值一致。模型在cuda1上训练完后，在cuda2上测试，模型失效了。结果：当cuda1上的x直接把值赋给cuda2上的x2时，x2的值全为零。
2024-05-29 18:07:37
220
1
原创 RuntimeError: cuDNN version incompatibility: PyTorch was compiled against (8, 9, 2)
安装pytorch时会安装cudnn相关的包，但是更换pytorch版本时不会删除cudnn的包，所以环境中会存在多个版本的cudnn，把不需要的包删除问题就解决啦。
2024-05-29 17:41:10
425
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人