在Jupyter Notebook中安装super_gradients解决YOLO-NAS报错及配置

1. 在Jupyter Notebook中安装super_gradients #

直接在Notebook的代码单元格中执行以下命令完成安装：

!pip install super-gradients

若要避免版本兼容问题，可指定适配ultralytics的最低版本：

!pip install super-gradients>=3.1.0

2. 额外需要的依赖模块 #

除super_gradients外，还需确保以下核心依赖正常：

• ultralytics：若版本过旧可能引发兼容问题，可更新至稳定版：

  !pip install --upgrade ultralytics
  

• torch与torchvision：YOLO-NAS基于PyTorch运行，CPU环境直接执行安装命令：

  !pip install torch torchvision
  

  若使用GPU，需安装与本地CUDA版本匹配的PyTorch包。

3. 验证安装 #

安装完成后重启Jupyter Notebook内核，重新运行你的代码即可：

from ultralytics import NAS
model = NAS('yolo_nas_s.pt')

内容的提问来源于stack exchange，提问作者Ирина Соловьева