win11搭建一个新的Pytorch环境，用来训练YOLOV5（win10也适用）

原创

已于 2022-10-26 11:18:12 修改 · 1.5k 阅读

18 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #python

于 2022-07-05 22:02:16 首次发布

本文分享了在Win11/Win10下搭建YOLOV5环境的过程，包括解决GPU调用失败及模型预测问题。环境配置为CUDA 10.2、cudnn 7.65、Python 3.8等。

我想搭建一个YOLOV5的环境，然后没想到一折腾就是两天，最后总算成功运行了，在这里我只想分享一下我的环境，希望能给那些被环境所困扰的人一个解决方案。

在这个过程中，我主要遇到了两个问题，一个是运行的时候无法调用GPU，报错跟CUDA有关系，另一个问题是YOLOV5能运行了，结果出来的模型预测没有框，mAP为零，百度了一下，还是环境的问题。

话不多说，直接分享我的环境和一些操作细节。

主要环境
我的电脑是Win11，这套环境Win11和Win10都能用，环境搭建用的Anaconda

CUDA=10.2
cudnn=7.65
python=3.8
torch=1.8.0+cu101
torchaudio=0.8.0
torchvision=0.9

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

qingfeng253

关注关注

1
点赞
踩
18

收藏

觉得还不错? 一键收藏
5
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

win10+yolov5+pytorch+gpu

m0_37556124的博客

09-17

1047

深度学习，yolov5训练自定义数据集

训练YOLOv11 | Pytorch和Pycharm的安装教程适用W11系统（附详细的系统变量添加过程）

热门推荐

Snu77的博客

10-01

1万+

根据你之前下载的版本选择对应的选项即可，然后复制底下的绿色代码到Anaconda的命令行输入即可(这里要在你安装的python环境里进行安装,例如我的环境名字是PyTorch,那么就是activate PyTorch里进行安装)然后同之前一样会加载一会，然后会询问我们是否下载配件,我们输入Y等待即可（下载过程中我们要确保网络通常如果有波动可能会下载失败）。类似于我的电脑是Windows11选择上面红框内的选项即可点击后即下载,(注意这里下载的是zip文件)下载完成后复制下面几个文件。

5 条评论您还未登录，请先登录后发表或查看评论

【YOLOv5】环境搭建：Win11 + mx450

我有两颗糖

06-22

3944

win11系统下，最新版 yolov5 的环境配置（详细）

PC机双系统（win11+ubuntu20.04）配置yolov5环境

weixin_48888831的博客

07-24

627

Win+R,输入msinfo32回车，出现系统界面，可以查看BIOS模式，确保为UEFI模式。

搭建Windows11下yolov5环境

LoongEmbedded的专栏

10-27

3572

Windows11下yolov5环境配置

配置YOLOv5环境并进行训练测试（Win11）

weixin_42506418的博客

06-30

1217

yolov5环境配置入门教程

在win10下进行pytorch版本yolov3安装及训练

12-22

权重文件：这个好久之前下载的了网址找不到了 2.代码环境 Readme写的是python3.7，我用的是3.6 其他的根据requirements.txt安装可运行，可以先使用以下代码安装 pip install -U -r requirements.txt 安装列表中包含 ...

Windows11下pytorch环境运行测试YOLOV5

weixin_43402278的博客

06-26

2311

python detect.py --source ./data/images/testfiles --weights ./weights/目标检测模型/yolov5x.pt。不过要注意，自己的数据集不建议放在这个路径下面，建议把数据集放到YOLOv5项目的同级目录下面。注意：安装依赖时要在自己创建的虚拟环境中安装，并路径要切到requirements.txt文件的项目路径中。以上就是搭建相关环境对yolov5进行运行测试，后续可以在yolov5模型的基础上做自己的数据集并进行训练。// 安装所需要的库。

基于windows10的pytorch环境部署及yolov8的安装及测试_yolov8环境搭建win10

2401_89791386的博客

01-15

1098

在这个base环境中先不要着急安装各种各样的python包，因为默认情况下在anaconda中创建的新的环境都是以base环境为模板的，也就是意味着新创建的环境会包含与base环境相同的Python版本和已安装的软件包列表，所以为了不必要的麻烦，我们新创建一个环境，创建环境的命令如下。1.下载代码，到官网去找，或者其他人上传的都可以，主要下载两个内容：下载地址：https://github.com/ultralytics/ultralytics。-n是名字的意思，name自己取名，后面那个3.8是版本号。

Windows11下yolov5环境配置系列教程（基础部分）

u010416569的博客

07-28

2190

condacreate-nXXX(环境名）condacreate-nXXX(环境名）python=X.X（该环境中你希望安装的python版本）安装过程中需要注意安装路径的选择，一般为默认。1.创建一个yolov5的python环境（在CMD或者AnacondaPrompt（Anaconda3）创建）以上仅仅是最基础的环境部分，希望对和我一样的小白有所帮助。3.我们可以进入该环境看看我们配置好的东西，输入activate你想进入的环境名。在你的下载目录下找到Anaconda3并点击。...

pytorch yolov5 推理和训练验证环境搭建

tugouxp的专栏

04-21

4484

关于之前pytorch框架下yolov3推理和训练环境搭建可以参考，本文所使用的环境和这篇记录一致。 pytorch yolov3 推理和训练环境搭建_papaofdoudou的博客-优快云博客_yolov3环境搭建下载代码框架环境： git clone https://github.com/ultralytics/yolov5 安装依赖： pip install -r requirements.txt 下载YOLOV5模型文件：下载链接：Releases · ultraly

(windows) yolov5/yolov3 pytorch环境配置gpu+ windows11+ anaconda+ pycharm+ RTX3050 笔记

chao_xy的博客

03-07

1728

(傻瓜式)十分钟完成YOLOv5环境搭建（Windows11系统）

qq_38702496的博客

12-25

988

下载pytorch，可去官网。查看适合自己显卡cuda版本。

Windows11下yolov5环境配置系列教程（进阶部分）

u010416569的博客

07-28

3840

而且每个人的问题也不尽相同，为了教程的简洁性，就不对遇到的问题展开了。我们可以看到，环境中虽然有cuda和pytorch，但是并没有cudnn，所以我们还要下载一个对应版本的cudnn。（这个cudatoolkit是包含在从官网下载的pytorch的包里的,所以不用另外下载一个CUDA）2.配置Yolov5所需要的环境,我们解压所下载的yolov5压缩包并打开。打开CMD，激活你所配置的环境,进入你解压好的yolov5文件路径。也可以通过我的百度云下载我正在使用的yolov5版本。...

超级详细Windows11部署YOLOV5并训练自己的数据集

最新发布

2301_81286570的博客

02-11

5803

完成后，运行python detect.py即可,在runs/detect/exp文件路径下，能够显示以下图片，表示yolov5部署成功。首先打开pycharm，新建项目，如下图所示，找到下载好的anaconda文件夹中python路径，接着点击创建。将下载好的cudnn文件夹中的include，bin，lib分别放到cuda对应的文件夹中。找到对应的cudnn版本，点击下载安装即可，下载完成后，解压到容易找到的路径。打开环境变量，注意，这次是在上面的栏目找到path，点击进行编辑。

PyTorch+Yolov5环境搭建

weixin_44332602的博客

10-11

2143

免额外安装cuda、cudnn

win11系统Python深度学习环境搭建与yolov5简单推理应用

2301_79528824的博客

07-10

1192

在同学指导下不求甚解的搭建了虚拟环境，简单的跑通了yolov5，这里简单记录一下本文纯自用，想学习的话还是另寻他处。

yolo11笔记（1）——安装pytorch与 Ultralytics（win/Mac/linux三平台）

weixin_45569617的博客

08-20

9592

电脑配置 cpu: i7-8750H gpu: 1060 6G 内存： 16G 操作系统：win10 anaconda 3.x 1. 安装pytorch 1.1 创建新的环境打开命令行输入创建一个新环境，并激活进入环境。 # 创建了名叫pytorch的，python版本为3.7的新环境 conda create -n pytorch python=3.7 # 激活名叫pytorch的环境 activate pytorch 1.2 下载pytorch 打开pytorch的官网https://pyt

Windows11下yolov5环境配置系列教程（问题解析部分）

u010416569的博客

08-05

1778

打开PyCharm，选择file-settings-project-project interpreter,如下图所示。如果打开之后没看到opencv-python 的package，则选择右上角的“+”号搜索opencv-python. 最后记得点击右下角的应用。安装完成的时候，会提示"Package ‘opencv-python’ installed successfully"可以看到，我们安装cudnn时报错了，这个错误可以更换源解决。在它给出来的一大堆版本中找到我们需要的版本。...

在Win10上使用YOLOV3-pytorch训练自定义数据集教程

- **操作系统：Win10**，YOLOv3-pytorch可以在Windows 10操作系统上运行，这使得它对Windows用户更加友好。 - **CUDA版本：9.0**，在训练和运行基于GPU的深度学习模型时，CUDA版本需要与PyTorch版本兼容，此处指的是...