torch-npu的配置+yolo

摘要

主要是torch安装和yolo训练的适配方法

1、简单的调用npu

import torch
import torch_npu
x = torch.randn(10000, 10000).npu()
y = torch.randn(10000, 10000).npu()
for _ in range(10000):
    z