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

  • 0
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值