一、前言
CenterNet有两个,本文的CenterNet对应的论文是:Objects as Points
对应Github地址:xingyizhou/CenterNet
Win10版本点击这里:【目标检测】Win10+CUDA10.0+CUDNN7.5搭建CenterNet环境
下面这两篇博客也对我帮助很大:
在SeaShips数据集上训练CenterNet网络
(绝对详细)CenterNet训练自己的数据(pytorch0.4.1)
本电脑环境为:
Ubuntu 16.04
GPU RTX2070 Advanced OC 8G
GPU驱动 418.87.00
gcc version 5.4.0
CUDA 10.0.130
CUDNN 7.6.0
python 3.6.9
pytorch 1.1.0
torchvision 0.3.0
二、关于CUDA版本
此时运行可视化代码和测试代码都是没有问题的:
可视化:
python demo.py ctdet --demo /home/vincent/Code/CenterNet/images/ --load_model /home/vincent/Code/CenterNet/models/ctdet_coco_dla_2x.pth
测试:
python test.py ctdet --exp_id coco_dla --keep_res --load_model ../models/ctdet_coco_dla_2x.pth
但是当我运行训练代码的时候就会报错:
训练:
python main.py ctdet --exp_id coco_dla --batch_size 5 --lr 1.25e-4 --gpus 0 --num_workers 0
报错内容的最后一行:
RuntimeError: cuda runtime error (11) : invalid argument at /opt/conda/conda-bld/pytorch_1535491974311/work/aten/src/THC/THCGeneral.cpp:663
完整报错内容点击这里
但是找了很久也没发现是什么原因,最后跟着Error for run demo.py #356换了个CUDA版本然后就可以了,所以目前用的是CUDA10.0
三、环境安装
1. 安装CUDA和CUDNN
1.1 下载
CUDA下载网址:CUDA Toolkit 10.0 Archive
两个文件都需要下载下来
CUDANN下载网址:cuDNN Archive
选择Download cuDNN v7.6.0 (May 20, 2019), for CUDA 10.0,下载cuDNN Library for Linux
1.2 安装CUDA
终端进入到CUDA文件地址:
首先安装基础包:
sudo sh cuda_10.0.130_410.48_linux.run
在安装过程中,会出现几个是否选装和安装位置的选项,其中第二个不需要