当前位置: 首页 > news >正文

YOLOv8的GPU环境搭建方法

        首先说明这个环境搭建教程是基于电脑已经安装好CUDA和CUDNN的情况下,去搭建能够正确运行YOLOv8代码的Pytorch的GPU版本。具体安装方法可见:最适合新手入门的CUDA、CUDNN、Pytorch安装教程_cuda安装-CSDN博客

        第一步:需要在cmd中创建虚拟环境conda create -n 环境名字 python版本,例如创建一个名字为YOLOv8的虚拟环境,并且指定Python版本为3.9,则需要在cmd中输入的代码为:

conda create -n YOLOv8 python==3.9

即可成功创建。

        第二步:激活指定的虚拟空间,例如激活名为YOLOv8的虚拟环境,其代码如下:

conda activate YOLOv8

就会进入对应的虚拟环境。

        第三步:在虚拟环境中安装运行YOLOv8官方代码所需要的模块。推荐首先安装ultralytics模块。如果下载较慢,可以添加镜像源。使用镜像源安装模块的方法可见:Python使用镜像安装模块_python安装镜像-CSDN博客

pip install ultralytics

        第四步:安装对应的CUDA的Pytorch版本,首先进入Pytorch官网,选择对应的版本进行pip安装,如下图:

但是在这里存在一个问题,安装很慢很慢,往往超时,我们可以将安装命令输入CMD中,会产生如下图:

复制此下载路径到迅雷中,粘贴到新建下载任务中,然后下载完成后,激活虚拟环境,然后进入到下载此文件的路径下,输入如下代码:

pip install “torch-2.4.1+cu118-cp39-cp39-win_amd64.whl”

 即可安装成功,但是在运行YOLOv8代码的时候,会报错torchvision与CUDA不适配,主要是在安装ultralytics模块的时候安装的torchvision是cpu版本的,torch + torchaudio都是coda版本,torchvision却不是,这就是问题所在,需要手动卸载torchvision,然后输入对应的cuda版本的torchvision安装,实例代码如下:

pip install torchvision==0.19.1  --index-url https://download.pytorch.org/whl/cu118

参考博客:[YOLOv8] 问题分析和解决 - NotImplementedError: Could not run ‘torchvision::nms‘ ...from the ‘CUDA‘ backend_yolov8notimplementederror: could not run 'torchvis-CSDN博客

完美解决下载安装pytorch cuda版本超级慢的问题_pytorch下载太慢-CSDN博客

http://www.lryc.cn/news/441465.html

相关文章:

  • JZ2440下载后设置NAND启动文件系统
  • AI绘画与摄影新纪元:ChatGPT+Midjourney+文心一格 共绘梦幻世界
  • 金手指设计
  • Chainlit集成LlamaIndex并使用通义千问模型实现AI知识库检索网页对话应用增强版
  • 详解c++菱形继承和多态---下
  • python学习笔记目录
  • 非结构化数据中台架构设计最佳实践
  • 鹏鼎控股社招校招入职SHL综合能力测评:高分攻略及真题题库解析答疑
  • 【测向定位】差频MUSIC算法DOA估计【附MATLAB代码】
  • 智能车镜头组入门(四)元素识别
  • Java键盘输入语句
  • 【读书笔记-《30天自制操作系统》-22】Day23
  • C++学习笔记(33)
  • 智谱清影 -CogVideoX-2b-部署与使用,带你揭秘生成6s视频的极致体验!
  • 探索Java中的设计模式:原则与实例
  • 【Java】关键字-static【主线学习笔记】
  • 数字自然资源领域的实现路径
  • GitLab邮箱发送邮件:如何实现自动化发信?
  • sqli-labs靶场搭建
  • Leetcode Hot 100刷题记录 -Day14(矩阵置0)
  • 每日刷题(算法)
  • 大牛直播SDK核心音视频模块探究
  • gin配置swagger文档
  • 基于ssm的快餐店点餐系统设计与实现
  • 集合框架底层使用了什么数据结构
  • Activiti7《第二式:破剑式》——工作流中的以柔克刚
  • docker快速搭建kafka
  • 基于 onsemi NCV78343 NCV78964的汽车矩阵式大灯方案
  • OpenAl o1论文:Let’s Verify Step by Step 快速解读
  • Errorresponsefromdaemon:toomanyrequests:Youhavereachedyourpullratelimit.