一 下载CUDA 和cuDNN 安装包
NVIDIA 官网 CUDA各版本 cuDNN各版本(需要注册账号才能下载)
注:CUDA和cuDNN对应版本如下
cuda_10.2.89_441.22_win10.exe
cudnn-10.2-windows10-x64-v7.6.5.32.zip
1.首先下载CUDA
CUDA 10.2下载地址:https://developer.nvidia.com/cuda-10.2-download-archive
选择深色,如果想在Linux上安装CUDA就选择Linux,Mac上就选择Mac。此处讲Win10安装教程
最后一个Installer Type有两个选项,前面是网络安装,后面是本地安装,网络安装就是下载一个很轻巧的一个工具包,大概20M,如果选择本地安装,就会将所有相关的东西都下载下来,安装过程中无需再次下载。
2.再下载cuDNN
cuDNN 10.2下载地址:https://developer.nvidia.com/rdp/cudnn-download
注:cuDNN下载时需要注册一个账号,才可以进入到下载界面。
进入下载界面
释放历史存档
看到更多版本
点击CUDA10.2下对应得windows10版本
二、Win10安装CUDA10及cuDNN
1.安装CUDA10.2
注:安装之前如果有其他版本的CUDA,为防止产生错误,请先卸载。不卸载会有两个版本的CUDA安装上
(同一版本重复安装会覆盖)
开始安装
注:标红的不要勾选,否则会安装失败
安装路径选择默认就好
等待安装完成
安装完后,检查环境变量
cmd验证安装
验证安装成功后,要配置环境变量,Path需要手动添加如下路径,对应上一步的安装路径
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp
2.cuDNN配置
将CUDA10.2对应的版本cuDNN文件:cudnn-10.2-windows10-x64-v7.6.5.32.zip解压
把cuDNN里面的三个文件夹复制到 CUDA10.2目录中 ,不用担心,一起复制,没有重复文件,不会把里面的某些文件替换掉。
一般CUDA默认安装在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
三、图片和视频的目标检测实现
1.下载及配置DarkNet
下载:
方式一:从百度云盘中下载配置好的DarkNet文件(可以跟着云盘中的视频梳理一遍)下载
方式二:从 https://github.com/AlexeyAB/darknet 下载,下载及配置过程见百度云盘下载
2.按照视频(How to build yolo on windows.mp4)及压缩包-darknet.zip(已经按照视频配置好了可以直接进行图片和视频的检测)
注:
图片检测输入命令:darknet.exe detector test cfg/coco.data cfg/yolov4.cfg yolov4.weights
视频检测输入命令:darknet.exe detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights test.mp4
(test.mp4也可在命令运行之后输入)
参考地址: