YOLOv5要求:Python》=3.7 pytorch》=1.7
1.驱动下载
下载旧版本英伟达驱动
关于何如在英伟达官网上下载历史驱动的方法_BeatifulBoyBoy的博客-CSDN博客_英伟达历史驱动
我选择的是419.17版本
2.CUDA及cudnn 安装
根据驱动版本(419.17)找到控制面板,找到组件查看cuda版本(我的是10.1),所以选择安装10.1版本。
点击进入CUDA Toolkit Archive网址,下载其中的cuda10.1版本。
cudnn根据cuda安装对应的版本,将解压的三个文件复制到cuda安装的文件中。