YOLO部署流程(新版本CUDA>11.8,DNF脚本前置)

一、查找最新驱动

https://www.nvidia.cn/geforce/drivers

二、首先判断电脑的cuda版本

#查看cuda版本 
nvidia-smi

三、对照cuda版本选择pytorch版本

较新版本:https://pytorch.org/

次新版本:https://pytorch.org/get-started/previous-versions/

cuda12.1:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

四、创建yolo的conda环境

conda env list
conda create --name y8 python=3.9 -y
conda activate y8

五、其他库

pip install ultralytics

#dnf部署
pip install pyserial
#需要复制文件到System32,文件所在路径envs\y8\Lib\site-packages\pywin32_system32\
pip install pywin32
pip install pygetwindow
pip install opencv-python
pip install pymysql

六、conda其他操作

#退出当前conda环境
conda deactivate 
#删除环境
conda remove -n y8 --all -y

Comments

No comments yet. Why don’t you start the discussion?

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注