在ubuntu18.04下训练yolov3模型（GPU）

在ubuntu18.04下训练yolov3模型（GPU）

1.安装NVIDIA显卡驱动

安装显卡驱动有添加ppa源和下载官方安装包的方法。

这里介绍一个相对简单点的方法为ubuntu 18.04安装显卡驱动：

（1）查看gpu和推荐的驱动版本：

$ ubuntu-drivers devices

（2）然后进行自动安装

$ sudo ubuntu-drivers autoinstall

在设置里面点击详细信息，如果显示出显卡信息就说明显卡驱动安装成功了。

2.安装cuda并配置cudnn

（1）对GCC降级

Ubuntu 18.04预装GCC的版本为7.3,而cuda编译需要更低的版本，这里对gcc进行手动降级至4.8版本。

对gcc降级：

sudo apt-get install gcc-4.8

对g++降级

sudo apt-get install g++-4.8

然后进入到/usr/bin目录下输入：ls -l gcc*

cd /usr/bin



ls -l gcc* 

显示结果如下：

 lrwxrwxrwx 1 root root 7th 3月  20 11:38 /usr/bin/gcc -> gcc-7.3  

表示gcc链接到gcc-7.3, 需要将它改为链接到gcc-4.8，方法如下:

sudo mv gcc gcc.bak #备份



sudo ln -s gcc-4.8 gcc #重新链接

同理， 将g++链接到g++4.8：

sudo mv g++ g++.bak



sudo ln -s g++-4.8 g++

在/usr/bin目录下查看gcc和g++版本