目标检测 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)
还没有任何评论哟~
