【踩坑】使用CenterNet训练自己的数据时的环境配置与踩坑

环境配置

系统：Ubuntu22.04
Python：3.8
CUDA：11.7
pytorch：2.2.1

因为种种原因没有使用原工程的老版本python和pytorch……但总之也跑通了，可喜可贺，可喜可贺。

1. 新建conda环境：

 conda create --name CenterNet python=3.8
 conda activate CenterNet

2. 安装CUDA
   使用nvidia-smi命令（前提是你安装了显卡驱动），可以看到显卡信息，其中右上角表示显卡最高支持的CUDA版本号，安装的CUDA版本不要超过这个版本。
   
   因为pytorch官网写了对应CUDA11.8和12.1两种版本的安装方式，所以图省事我的CUDA版本也直接安装11.8了。
   首先找到你要安装的CUDA ToolKit版本：https://developer.nvidia.com/cuda-toolkit-archive
   
   然后按你的机器配置一步一步选择好，运行给你的指令：
   
   接下来的安装选项可以参考这篇博客，以下是一些细节：
   如果你安装了其他版本的CUDA，会报一个询问是否continue的链接，continue即可。
   注意在安装选项处去掉Driver、CUDA Demo Suite 11.8、CUDA Documentation 11.8，仅安装CUDA Toolkit 11.8。
   使用如下命令设置CUDA环境变量：

export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/us