多GPU训练的模型,转换加载到单GPU测试或预训练
多GPU训练的模型,转换加载到单GPU测试或预训练
报错:RuntimeError: Attempting to deserialize object on CUDA device 3 but torch.cuda.device_count() is 1.
加载的模型是3个GPU进行训练的,而你的计算机只有1个GPU
state_dict = model.state_dict()
weights_path = torch.load(opt.weights_path)['model_sta
原创
2021-08-02 22:04:58 ·
1790 阅读 ·
0 评论