基于Python3.10的YOLO-NAS环境配置

1.创建虚拟环境

conda create -n 环境名称 python=版本名称

2.检查电脑CUDA版本

#输出结果为显卡性能图，右上角可以查看CUDA版本
nvidia-smi

3.安装兼容对应版本CUDA的Pytorch

#官方网站 https://pytorch.org/
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu版本号

4.安装YOLO-NAS所需的其他库 

pip install numpy

pip install sympy

pip install cython

pip install imutils

pip install roboflow

pip install pytube --upgrade

5.安装super-gradients库

#官方网站 https://pypi.org/project/super-gradients/
pip install super-gradients==与Pytorch匹配的版本号

 6.检查CUDA与Pytorch版本兼容性

#版本查询
import torch
torch.__version__

#兼容性查询
import torch
torch.cuda.is_available()