一、前言
CenterNet有两个,本文的CenterNet对应的论文是:Objects as Points
对应Github地址:xingyizhou/CenterNet
其实Win10安装过程和Ubuntu方法差不多的,本文主要说说本人在Win10上安装过程中需要注意的地方。
Ubuntu版本点击这里:【目标检测】Ubuntu16.04+RTX2070+CUDA10+pytorch1.1搭建CenterNet环境
本电脑环境为:
Win10
GPU TITAN RTX
GPU驱动 430.86
CUDA 10.0.130
CUDNN 7.5.0
python 3.6.9
pytorch 1.1.0
torchvision 0.3.0
二、CUDA和CUDNN
1.1 下载CUDANN
CUDA下载网址: