pytorch分布式RuntimeError: Tensors must be CUDA and dense

阅读量：

你的模型或参数未加载至GPU中，需采取相应措施进行处理：

复制代码

    backbone = eval("backbones.{}".format(args.network))(False, dropout=dropout, fp16=args.fp16).to(rank)
    
    
    bash

最终的 .to(rank) 操作实现了这一目标。在本人所编写的代码中，rank 的取值设定为 0

全部评论 (0)

还没有任何评论哟~