【目标检测】Ubuntu16.04+RTX2070+CUDA10.0+pytorch1.1搭建CenterNet环境

一、前言

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文件地址：
首先安装基础包：