未来农业科技

 找回密码
 立即注册
查看: 7231|回复: 0

Yolo_v8_windows11 下的环境搭建

[复制链接]

5

主题

5

帖子

37

积分

版主

Rank: 7Rank: 7Rank: 7

积分
37
发表于 2024-2-29 15:57:10 | 显示全部楼层 |阅读模式
本帖最后由 ferret 于 2024-2-29 15:59 编辑

Yolo_v8_windows11 下的环境搭建环境准备
cuda、cudnn、pytorch、python3.8(极力推荐)、pycharm
cuda、cudnn、pythorch 版本选择查看本机支持 cuda 最高版本(cuda 向下兼容)
找到 NVIDlAControlPanel 控制面板
系统信息>组件>NVCUDA64.DLL
如图所示,我的最高支持 12.3 版本
所以我需要安装 12.3 以下版本的 cuda
查看 pytorch 最高支持的 cuda 版本
pytorch 官网

如图所示,截止写稿。pytorch 支持 12.1 和 11.8 的 cuda 版本
因为我的是 12 系列所以选择,12.1 版本的 cuda
查看 cudnn 最高支持的 cuda 版本
cudnn 下载网站

注意版本区别

我选择的是 12.1 版本的 cuda,所以这里选择 12.x 的最新版
结论:cuda:12.1.x | cudnn:8.9.7 | pytorch:使用官网链接安装pycharm 和 python3.8
略,自行百度
安装安装cuda
打开下载好的cuda安装包,选择解压目录(我用的默认路径)后解压:

同意许可条款后:选择自定义安装

全选:

安装位置默认(极力推荐):

勾选后下一步:

等待安装完成后即可。可能需要重启。
环境变量(默认已经配置好,无需操作)
快捷键:win+r,输入
sysdm.cpl


安装cudnn:
解压下载的cudnn包后:全选、复制/剪贴

找到cuda安装目录(默认位置):
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1

在这个目录下粘贴,替换所有。安装完毕。
验证安装是否成功:
cmd输入:
nvcc -V
出现如下内容即表示成功!
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Archiver|手机版|小黑屋|未来农业科技技术学习 ( 赣ICP备2022001303号 )|

GMT+8, 2024-12-27 00:21 , Processed in 0.014368 second(s), 20 queries .

Powered by Discuz! X3.4

Copyright © 2001-2020, Tencent Cloud.

快速回复 返回顶部 返回列表