yolov5-pytorch环境安装教程

yolov5-pytorch环境安装教程

一． 准备工作：Cuda 、Anaconda安装、yolov5源码下载

1.cuda版本查看：
如果没有下载cuda的话，自行百度即可，例如看B站up主：我是土堆的视频，我也是按照他的来装的，这是视频链接。

查看本机电脑cuda版本的方法：

方法：win+R打开cmd，输入nvidia-smi，效果如下

2.annacoda安装：

使用百度等引擎搜索anaconda进入官网之后下载即可（建议使用安装包，省得自己再去配置环境变量，麻烦，一步到位）按照网上的教程一直安装就好了。
例如B站的我是土堆up主：链接，链接2

附上Anaconda3-5.2.0-Windows-x86_64版本的安装包
百度云链接：点击链接获取，提取码：1111

创建pytorch环境：
安装annaconda完之后，我们需要新建pytorch环境，（annaconda安装完成之后会自带一个base环境，里面包括了一些常用的python第三方库类，但并不包括torch，因此我们需要建立一个带有torch的python编译环境，当然，你也可以选择用base环境来安装torch，不过安装时要注意base环境下的python版本需要与torch版本对应）

打开Anaconda Prompt执行代码：

# 创建了名叫pytorch，python版本为3.7的新环境
conda create -n pytorch python=3.7  

3.下载YOLOv5 github项目
点击前往yolov5GitHub链接地址

把压缩包解压到指定目录就行。

二． 安装pytorch及yolov5相关第三方库

1.pytorch安装：
如果你的电脑已经安装了cuda，可以按照你自己的cuda版本下载pytorch对应版本的离线whl文件进行即可。
推荐使用离线安装，在线太慢了。。。。。。
torch相关下载地址：https://download.pytorch.org/whl/torch_stable.html
打开网页后CTRL+F 快速搜索需要安装的torch版本
比如：我的cuda版本是11，创建pycorch环境的python版本是3.7则搜索：
ctrl+f 输入cu11然后找到对应的cp37（cp37即python3.7）选择合适版本点击下载即可

这里是我用的对应的torch、torchvisio的版本，我是2022年10月31日正常通关的，有CPU版本和GPU版本，自己在链接里找适应自己的，点击链接获取，提取码：1111
2.打开cmd执行代码进行安装即可：

# 打开cmd或者Anaconda Prompt激活名为pytorch的环境
activate pytorch
# 跳转到安装好的whl文件目录下
cd C:\Users\17394\Downloads
# pip安装下载好的whl文件
pip install torch-1.7.1+cu110-cp37-cp37m-win_amd64.whl
pip install torchvision-0.8.2+cu110-cp37-cp37m-win_amd64.whl

3.安装yolov5所需的第三方库：
yolov5项目下载解压后有一个requirements.txt文件

继续执行以下代码安装即可：

pip install requirements.txt

三、配置pycharm打开yolov5项目
1.打开yolov5项目修改python解释器
CTRL+ALT+S 打开设置
2.执行以下步骤

到此yolov5的环境就已经搭建好了，我是根据下面链接中博主的文章，加上我直接给了各位安装包，我试了试，这个torch和torchvision的版本是可以跑yolov5.5.0版本的代码，下一个文章我将写一下我遇到的问题和解决方法，以及怎么搜解决方法
————————————————
版权声明：本文为CSDN博主「m0_45696735」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/m0_45696735/article/details/118253942