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()