目录
一、环境配置Pytorch+CUDA+cuDNN
YOLO的环境配置真的异常繁琐,刚上大学那会儿很多次想学各种深度学习,几乎每一次都倒在环境配置上。
1. 安装CUDA和cuDNN
1.1 查看显卡驱动对应的CUDA版本
建议在把显卡驱动更新到最新。
好用的显卡真的非常重要!GTX850M和RTX3080比起来真的天壤之别。
要安装现版本yolov5就要安装对应的版本的pytorch(我记得是1.9版本以上),而该版本的pytroch对应的CUDA(我记得好像的11.1)对应的显卡驱动相当新,几乎无法避免地要更新显卡驱动(甚至换一张好的显卡!)
右击桌面可以查看英伟达驱动设置-》帮助-》组件