解决RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1
解决方法，在加载模型的语句上，增加map_location='cuda:0'。
·
我是在服务器上训练的模型，然后下载到本地进行测试，报错如下：
问题应该是训练和测试的GPU不是同一个，需要将GPU改成本地的。
解决方法，在加载模型的语句上，增加
map_location='cuda:0'
欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
1
0- 0
已为社区贡献1条内容
所有评论(0)