在Ubuntu环境下跑原生darknet框架的yolov4进行绝缘子识别时，首先需要安装Nvidia显卡驱动，并配置对应的CUDA和cudnn环境。

1 cuda安装

根据显卡驱动版本安装合适的cuda版本。注意cuda可以多版本共存，并通过软链接的方式进行多版本切换。

比如，我们安装了10.2版本的cuda，默认会安装在/usr/local/cuda-10.2/下，/usr/local/cuda/文件夹通过软链接挂载到/usr/local/cuda-10.2/。我们可以通过改变/usr/local/cuda/的软链接挂载位置实现cuda多版本切换。

stat命令查看当前cuda软链接指向的哪个cuda版本：

  1. stat cuda

删除原来的软链接，重新挂载：

sudo rm -rf usr/local/cuda                      
sudo ln -s /usr/local/cuda-10.2 /usr/local/cuda  #注意cuda后不要加/

2 cudnn安装

根据cuda版本安装合适的cudnn（注意 需要登录Nvidia官网才能下载）。

将cudnn文件夹中的对应文件copy到/usr/local/cuda-10.2/（或其它版本），即可完成cudnn安装。