在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目录下查看