yoloV5的环境安装

102 阅读1分钟

一. python环境准备

我这边使用的是python3.8和Anaconda管理的环境

conda.jpg

二.pytorch安装

  1. 当前的电脑显卡是英伟达RTX4060,电脑是联想拯救者Y7000p, 安装的cuda版本是111,pytorch1.8.2可以正常使用

cu111.jpg

安装命令: 
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官网可以看看自己电脑支持什么样版本的

pytorch.jpg

三. 拉取代码安装其他环境

  1. 使用git clone拉取github.com/ultralytics…
    拉下代码后使用

    pip install -r requirements.txt
    
  2. 安装完成后,python detect.py执行监测命令,如果正常运行,环境就安装成功了