Yolov8环境设置(一)windows环境

730 阅读1分钟

环境安装配置

1. 安装虚拟环境Conda

下载安装anaconda:

初始化conda环境

  • conda create -n <环境名称> python=<版本号> (python版本>=3.8)

激活当前环境

  • conda activate <环境名称>

  • 查看Pytorch版本 pip list 企业微信截图_16787013218917.png

2. 安装CUDA

命令行输入 nvidia-smi 查看CUDA支持的最高版本 (仅支持Nvidia显卡) 企业微信截图_16787018062961.png

为了确保版本兼容,请先删除nvidia所有的文件

企业微信截图_16787022789771.png

下载安装版本CUDA

企业微信截图_16787020046647.png

企业微信截图_16787021146284.png

下载安装cuDNN,选择对应的版本:(如:我是CUDA 11.6,选择CUDA 11.X的均可,建议不要选最新版本以免引起兼容问题)

企业微信截图_16787027292175.png

拷贝这三个文件

企业微信截图_16787027867058.png

粘贴到此目录

企业微信截图_16787028415216.png

3. 安装PyTorch (命令需要包含+cu字样)

Previous PyTorch Versions | PyTorch

企业微信截图_16787734794024.png

企业微信截图_16787734254845.png

查看pytorch是否安装成功 pip list

企业微信截图_16787735506702.png

下一章节讲解如何使用ultralytics(Yolov8)