【yolov8】3、yolov8 环境安装 【GPU版】

1、要求

  • GPU版的cuda需要10.2以上版本

1.1 什么是 CUDA 和 CUDNN

  • CUDA 是 NVIDIA 推出的一款通用并行计算平台和编程模型,用于 GPU 加速的计算。它允许开发人员利用 GPU 的强大计算能力来加速各种应用程序,包括科学计算、机器学习、图像处理和视频处理。

  • cuDNN 是 NVIDIA 开发的基于 CUDA 的深度学习加速库,为深度学习框架提供高效的 GPU 加速。它包含了常用的深度学习卷积、池化、归一化和激活等操作的 GPU 实现,可以显著提高深度学习模型的训练和推理速度。

1.2 查看cuda版本的3种方法(版本在10.2以上的可以忽略本章节):

  1. 右键电脑右下角,选择 NIVDIA控制面板
  2. 右键电脑桌面空白处,选择NIVDIA显卡控制面板
  3. 电脑左下角开始菜单,输入NIVDIA,选择它
  • 帮助 → 系统信息 → 组件 → CUDA

在这里插入图片描述

1.3 没有找到NIVDIA图标,确认是否有英伟达显卡

  • 在电脑的管理,查看是否有英伟达显卡

  • 使用下面的网址,更新显卡驱动

  • 如果没有英伟达显卡,可以安装CPU版

Cuda下载地址:https://developer.nvidia.com/cuda-toolkit-archive
Cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-download

2、pycharm下载安装

进入官网

  • PyCharm Professional 专业版

  • PyCharm Community Edition 社区版(免费-学习 yolo 足够)

  • 推荐安装在非C盘,勾选所有,一直下一步

在这里插入图片描述

3、yolov8 环境安装 【GPU版】

3.1 Yolov8 源码下载:

  • 推荐,做了优化】蓝奏云:https://zqj.lanzouj.com/i3wgh1jpzpta

  • 官网版:https://github.com/ultralytics/ultralytics

Stage 1 :解压后用pycharm打开
Stage 2 :配置 python 解释器(anaconda)
  • 右下角 Add New lnterpreter - > Add Local interpreter... - > 添加上一节安装的 anaconda

  • 点击加载环境 Load Environments

在这里插入图片描述
在这里插入图片描述

  • 右下角出现yolov8表示配置成功,等待几分钟加载依赖即可

在这里插入图片描述

Stage 3 :安装Yolov8的依赖环境
Stage 3-1 :把 requirements.txt里面的torchtorchvision前面的加上#号注释

在这里插入图片描述

Stage 3-2 :【安装依赖】然后在终端运行安装代码: pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
-i 表示引用下载源,后面跟上具体下载源的地址。除了给出的清华源还有其他的,具体看下一页(可替换)
http://pypi.douban.com/simple/ 豆瓣
https://pypi.mirrors.ustc.edu.cn/simple/ 中国科学技术大学
http://pypi.sdutlinux.org/simple/ 山东理工大学
http://pypi.hustunique.com/simple/ 华中理工大学
http://pypi.mirrors.opencas.cn/simple/ 中国科学院
https://pypi.tuna.tsinghua.edu.cn/simple/ 清华大学
http://mirrors.aliyun.com/pypi/simple/ 阿里

在这里插入图片描述

Stage 3-3 :安装GPU版 pytorch
11.3(推荐):
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch –y

10.2:
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch –y

在这里插入图片描述

Stage 4 :验证GPU是否安装成功
Python
import torch
print(torch.cuda.is_available())

在这里插入图片描述

Stage 5 :测试
  • 打开 wzq.py,运行

在这里插入图片描述

  • 效果如下

在这里插入图片描述

在这里插入图片描述

相关推荐

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-20 17:26:01       67 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-20 17:26:01       72 阅读
  3. 在Django里面运行非项目文件

    2024-07-20 17:26:01       58 阅读
  4. Python语言-面向对象

    2024-07-20 17:26:01       69 阅读

热门阅读

  1. MySQL——视图

    2024-07-20 17:26:01       21 阅读
  2. Window任务栏应用图片无法加载解决方法

    2024-07-20 17:26:01       23 阅读
  3. linux shell(上)

    2024-07-20 17:26:01       23 阅读
  4. RK3588 编译opencv&opencv_contrib记录

    2024-07-20 17:26:01       27 阅读
  5. 二叉树---路径总和

    2024-07-20 17:26:01       20 阅读
  6. windows 安装 kubectl 并连接到 k8s 集群【图文教程】

    2024-07-20 17:26:01       22 阅读
  7. computeIfAbsent 和 putIfAbsent

    2024-07-20 17:26:01       22 阅读
  8. 微软Edge浏览器全解析教程

    2024-07-20 17:26:01       20 阅读
  9. 如何使用unittest框架来编写和运行单元测试

    2024-07-20 17:26:01       21 阅读
  10. 数学建模熵权法

    2024-07-20 17:26:01       24 阅读