Win10 搭建 YOLOv8 运行环境(20240423)

一、环境要求

        1、Python,版本要求>=3.7

        2、PyTorch,版本要求>=1.7。PyTorch 是一个开源的深度学习平台,为人工智能研究提供了一个灵活的、易于使用的工具集。YOLOv8 是基于 PyTorch 框架实现的,所以需要安装 PyTorch。

        3、CUDA和cuDNN(可选),如果要使用 GPU 进行训练的话需要安装。其中 CUDA 是由显卡厂商NVIDIA推出的运算平台与编程模型,它允许程序员利用NVIDIA GPU的并行计算能力,加速各种计算密集型应用程序;而 cuDNN 是 NVIDIA 推出的一个用于深度学习的 GPU 加速库,全称是NVIDIA CUDA 深度神经网络库 。它的主要功能和优势在于提供了一系列针对深度学习的高性能原语,如卷积、池化、归一化等。这些原语经过高度优化,能够充分利用 GPU 的并行计算能力,从而显著加速深度学习的计算过程。

二、环境安装

        为了避免不同项目之间的依赖包产生冲突、相互影响,推荐使用 Anaconda 来管理 Python 环境。Anaconda3 是一个开源的 Python 发行版,旨在简化包管理和部署数据分析环境。Anaconda3 允许用户轻松创建和管理多个独立的 Python 环境,以便在不同的项目中使用不同版本的Python和不同的库;Anaconda3 还集成了一个强大的包管理器,用户可以通过简单的命令行操作来安装、更新和卸载各种Python库,从而方便地管理自己所需的库。所以先从 Anaconda3 的安装开始记录。

        1、安装 Anaconda3

        1.1、下载

                官网地址:Distribution | Anaconda

                阿里云盘:阿里云盘分享

        1.2、 安装

                推荐默认安装,一路 Next,唯一需要注意的就是在第五步时把添加环境变量的选项勾选上,后期自己配置麻烦且容易出现难以排查的错误。

        2、安装 Python 3.11.9

        2.1、在开始菜单中找到并打开 Anaconda 的命令行工具,Anaconda Powershell Prompt 或者 Anaconda Prompt 均可。

               

        2.2、输入命令创建虚拟机环境,在创建虚拟机的同时也会下载安装 Python 3.11.9。

conda create –n yolov8 python=3.11.9

        2.3、激活虚拟机环境。

conda activate yolov8

        2.4、查看当前环境列表,确认虚拟机是否激活(即查看 * 号在哪个虚拟机环境上)。

conda env list

        3、安装 PyTorch

        3.1、在第 2 步中打开的 Anaconda 命令行工具中输入命令安装。

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

         3.2、这个命令是在 PyTorch 官网根据机器的实际情况自动生成的。

        3.3、需要注意的是 CUDA 的选择,这个要根据你有没有 GPU 以及 CUDA 所支持的最高版本来决定。不使用 GPU 的直接点击 CPU 选项,使用 GPU 则需要查询机器所支持的 CUDA 最高版本。可以通过 Windows 自带的 CMD 命令行工具输入命令进行查询。

nvidia-smi

        3.4、 最后可以通过 Anaconda 命令行工具查看已安装依赖来查看是否安装成功。

 4、安装 CUDA、cuDNN

        4.1、下载 CUDA

                官网地址:CUDA Toolkit 12.0 Downloads | NVIDIA Developer

                阿里云盘:阿里云盘不支持 CUDA 文件格式的分享,就没法提供了

        4.2、安装 CUDA

                推荐默认安装,一路"下一步"。如果是自定义安装的话记住自定义安装路径,配置环境变量和安装 cuDNN 会需要用到。

        4.3、可以通过 Windows 自带的 CMD 命令行工具输入命令 查看是否安装成功

nvcc -V
或者
nvcc --version

        

         4.4、下载 CUDA 对应版本的 cuDNN

                官网地址:cuDNN Archive | NVIDIA Developer

                阿里云盘:阿里云盘不支持 cuDNN 文件格式的分享,就没法提供了

         4.5、解压压缩包,复制 bin、include、lib 三个文件夹

         4.6、找到 CUDA 的安装目录,如果是默认安装的话应该在 “C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v安装的版本号”,将复制的三个文件夹粘贴到这个位置

由于篇幅原因,Python 集成开发环境 PyCharm 的安装,以及编写一小段代码调用测试 YOLOv8 提供的 API 将在本地运行测试 YOLOv8(20240423)-CSDN博客记录。 

相关推荐

  1. Win11+DockerCUDA开发环境

    2024-04-24 17:58:01       48 阅读
  2. QEMU_v8OP-TEE运行环境

    2024-04-24 17:58:01       16 阅读
  3. Hadoop运行环境

    2024-04-24 17:58:01       31 阅读

最近更新

  1. TCP协议是安全的吗?

    2024-04-24 17:58:01       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-04-24 17:58:01       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-04-24 17:58:01       18 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-04-24 17:58:01       20 阅读

热门阅读

  1. c++11引入的constexpr关键字

    2024-04-24 17:58:01       13 阅读
  2. Conda的使用:创建一个虚拟环境

    2024-04-24 17:58:01       11 阅读
  3. 偏差数据比对

    2024-04-24 17:58:01       13 阅读
  4. WordPress采集插件如何选择?哪里免费获取?

    2024-04-24 17:58:01       12 阅读
  5. Flask + Bootstrap vs Flask + React/Vue:初学者指南

    2024-04-24 17:58:01       11 阅读
  6. 算法小白刷力扣 1 - 两数之和

    2024-04-24 17:58:01       13 阅读
  7. linux-mysql安装

    2024-04-24 17:58:01       13 阅读
  8. vue实现进入某个页面后替换地址栏路径

    2024-04-24 17:58:01       14 阅读