PyTorch启用苹果gpu

[xian-zh]

     ##torch.py
        self.use_gpu = False
        if cfg.engine_cfg.use_cuda:
            self.device = torch.device(f"cuda:{cfg.engine_cfg.gpu_id}")
            self.predictor.to(self.device)
            self.use_gpu = True
        # Mac 上启用 PyTorch GPU 加速需通过 Metal Performance Shaders (MPS) 后端实现
        if cfg.engine_cfg.use_mps:
            self.device = torch.device("mps" if torch.backends.mps.is_available() else "cpu")
            self.predictor.to(self.device)
            self.use_gpu = True    
     ##use
     engine = RapidOCR(
          params={
              "Det.engine_type": EngineType.TORCH,
              "Cls.engine_type": EngineType.TORCH,
              "Rec.engine_type": EngineType.TORCH,
              "EngineConfig.torch.use_mps": True,  # 使用torch GPU版推理
          }
    
    )

[SWHL]

请问这样可以调用吗？

[xian-zh]

我这边运行是可以的，具体有没有使用gpu我也还没验证

[zzzhouuu]

只是用了torch引擎，并没有启用mps加速的

[SWHL]

请问这个怎么确定是否启用了mps加速呀？

[zzzhouuu]

当前代码除了显示指定use_cuda、use_npu下都运行在cpu上，我在mac上跑过做OCR时候gpu和显存并没有明显占用，倒是自己写的代码将device指定mps会有明显的调用。可以打印一下张量的 .device 属性就知道运行在什么设备上了

[SWHL]

好的。这个我来优化一下

[SWHL]

转到 issue了，后续会调研跟进：#572