NVIDIA+Win10+CUDA+cuDNN 及YOLO实现目标检测

一 下载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也可在命令运行之后输入)

参考地址:

【NVIDIA】Win10 + CUDA10 + cuDNN 安装教程

nvidia驱动,cuda与cudnn的关系

nvidia驱动,cuda,cudnn关系

YOLO框架简述

Yolo神经网络搭建、训练、PYthon调用YOLO.DLL、C++调用YOLO.DLL

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值