一. python环境准备
我这边使用的是python3.8和Anaconda管理的环境
二.pytorch安装
- 当前的电脑显卡是英伟达RTX4060,电脑是联想拯救者Y7000p, 安装的cuda版本是111,pytorch1.8.2可以正常使用
安装命令: pip install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111
2. 登录pytorch官网可以看看自己电脑支持什么样版本的
三. 拉取代码安装其他环境
使用git clone拉取github.com/ultralytics…
拉下代码后使用
pip install -r requirements.txt
安装完成后,python detect.py执行监测命令,如果正常运行,环境就安装成功了