本帖最后由 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出现如下内容即表示成功!
|