【目标检测】Win10+TITAN RTX+CUDA10.0+pytorch1.1搭建CenterNet环境

本文详细介绍了在Win10系统上,使用TITAN RTX显卡和CUDA10.0、CUDNN7.5.0,如何搭建用于目标检测的CenterNet环境。内容包括CUDA和CUDNN的下载与安装,解决pytorch安装、python库下载慢、EOFError、找不到pycocotools以及编译DCNv2等问题的步骤与技巧。
摘要由CSDN通过智能技术生成
一、前言

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下载网址:

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值