yolo-nas使用教程

创建yolo-nas环境

conda create -n yolonas python=3.8
# 安装yolo-nas包
pip install super-gradients
1
2
3

导入包并设置模型参数

import torch
from super_gradients.training import models
1
2

device = torch.device("cuda:0") if tor