Advertisement

目标检测 Ubuntu 16.04 + RTX 2070 + CUDA 10.0 + PyTorch 1.1 配置 CenterNet 部署环境

阅读量:

一、前言

【CenterNet存在两个版本,与本文所采用的CenterNet相对应的文献为:Objects as Points
其对应的GitHub项目地址为:xingyizhou/CenterNet

适用于Windows 10系统的下载链接为:【目标检测】Win10+CUDA10.0+CUDNN7.5搭建CenterNet环境

以下两篇博客在研究过程中提供了重要参考:
在SeaShips数据集上训练CenterNet网络
(绝对详细)CenterNet训练自己的数据(pytorch0.4.1)

当前所使用的计算机配置如下:

操作系统为Ubuntu 16.04
显卡型号为RTX2070 Advanced OC,显存容量为8G
显卡驱动版本为418.87.00
GCC编译器版本为5.4.0
CUDA工具包版本为10.0.130
CUDNN库版本为7.6.0
Python解释器版本为3.6.9
PyTorch框架版本为1.1.0
TorchVision库版本为0.3.0

二、关于C

全部评论 (0)

还没有任何评论哟~