新手可直接复现：Duan版本CenterNet在2080Ti上训练自己的数据集——踩坑记录

置顶

DJ_SIGS

于 2020-08-16 18:04:10 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏：目标检测文章标签：深度学习计算机视觉人工智能机器学习 python

本文链接：https://blog.youkuaiyun.com/NCEPUDJ/article/details/108038586

实验室换了GPU，记录下centernet在新电脑上的配置过程，如果对下面很多不理解先阅读我以前的这篇博客即可初学者复现CornerNet：详细指导零基础在Ubuntu系统运行该代码并完全理解论文思路的教程

直接复现

CenterNet-duan-2080Ti代码
按照 README上面的操作即可

踩坑记录

1. cuda版本问题

/home/psdz/anaconda3/envs/CenterNet/lib/python3.6/site-packages/torch/cuda/__init__.py:114: UserWarning: 
	 Found GPU0 GeForce RTX 2080 Ti which requires CUDA_VERSION >= 9000 for
	 optimal performance and fast startup time, but your PyTorch was compiled
	 with CUDA_VERSION 8000. Please install the correct PyTorch binary
	 using instructions from http://pytorch.org
	 ...
	 ...
	 ...
	   File "/home/psdz/anaconda3/envs/CenterNet/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward
self.padding, self.dilation, self.groups)
RuntimeError: CuDNN error: CUDNN_S