ubuntu18.04搭建yolov4训练环境_ubuntu18.04训练yolov4-tiny

环境配置(根据自己电脑来)：NVIDIA 11.4+cuda 11.4+cuDNN 8.2

1.安装NVIDIA显卡驱动

1.1查看电脑显卡型号
在ubuntu终端输入lshw -c video
根据product即可知道型号为GP107M [GeForce GTX 1050 Mobile]
1.2查找相应显卡驱动并下载
到网站https://www.nvidia.com/Download/index.aspx?lang=en-us查找，以我电脑为例，下载以下相应版本
1.3禁用nouveau
终端输入

sudo gedit /etc/modprobe.d/blacklist.conf
1

在blacklist.conf文件末尾添加以下语句并保存

blacklist nouveau
options nouveau modeset=0
1
2

终端输入，应用更改sudo update-initramfs -u
重启系统，验证是否禁用

lsmod | grep nouveau
1

若没有返回值则表示禁用成功
1.4安装NVIDIA驱动
打开系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改
查看是否安装成功nvidia-smi

2.安装cuda

2.1选择相应的版本并下载
在https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html查看cuda和对应的驱动
在https://developer.nvidia.com/cuda-downloads下载对应的cuda
2.2安装cuda
在下面的Bse Installer有相应的安装指令

wget https://developer.download.nvidia.com/compute/cuda/11.4.1/local_installers/cuda_11.4.1_470.57.02_linux.runsudo 
sh cuda_11.4.1_470.57.02_linux.run
1
2

continue—>accept回车—>Driver空格取消（不安装cuda附带驱动）—>Install回车 等待安装完成
2.3配置cuda环境
终端输入

sudo su 
sudo gedit ~/.bashrc 
1
2

在 ~/.bashrc 的末尾加上

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
1
2

更新环境source ~/.bashrc
检验是否安装成功nvcc -V

3.安装cuDNN

到官网下载https://developer.nvidia.com/cudnn，需要先注册登陆后使用
解压tar xvzf cudnn-11.4-linux-x64-v8.2.2.26.tgz
解压后可以看到cuda文件夹，在当前目录打开终端，然后将库和头文件copy到cuda目录并修改文件访问权限

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
1
2
3
4

查看cudnn版本验证是否安装成功

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
1