Advertisement

目标检测 Win10+TITAN RTX+CUDA 10.0+PyTorch 1.9 配置中心网环境

阅读量:

一、前言

【CenterNet存在两个版本,本文所涉及的CenterNet对应的研究文献为:Objects as Points
相关的GitHub项目地址为:xingyizhou/CenterNet

实际上,Windows 10系统的安装流程与Ubuntu系统大体相似,本文主要阐述在Windows 10系统安装过程中需要注意的事项。

如需获取Ubuntu系统的相关信息,请点击此处:【目标检测】Ubuntu16.04+RTX2070+CUDA10+pytorch1.1搭建CenterNet环境

当前使用的计算机环境如下:

Windows 10操作系统
显卡型号为TITAN RTX
显卡驱动版本为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软件的获取链接为:[CUDA Toolkit 10.0 Arc

全部评论 (0)

还没有任何评论哟~