1.安装CUDA 9.0
2018.07.29:已经有9.2版本但是暂时不建议安装
通过runfile安装
官方下载地址:https://developer.nvidia.com/cuda-90-download-archive
按如下选择,把安装包和补丁包都下了
检查显卡驱动版本
打开Driver Manager,选择NVIDIA的驱动
我的驱动是装系统的时候自动就装好了,版本号:390.48,这个版本号必须要比你下载的cuda安装文件cuda_9.0.176_384.81_linux.run里面的384.81版本高,不然安装失败.
如果没有驱动或者驱动版本低,则去官网下载安装
https://www.nvidia.com/Download/index.aspx?lang=cn
接下来就可以开始安装CUDA了
停止xserver,装驱动的时候也会有类似的提示
service lightdm stop
这会关闭桌面GUI,屏幕会黑屏,然后按ctrl + alt + f1切换console,输入用户名密码登录
给runfile添加执行权限
chmod +x ./cuda_9.0.176_384.81_linux.run
安装
./cuda_9.0.176_384.81_linux.run
有个是否需要安装驱动,选择 no
这样就完成了CUDA的安装
2.安装opencv
2018.07.29:到现在为止YOLOV3支持的最高版本为opencv 3.4.0
不推荐源码编译安装,跟CUDA9.0一起的时候有太多的坑
直接通过apt安装,版本为3.2
sudo apt install libopencv-dev
3.安装YOLO
官网:https://pjreddie.com/darknet/yolo/
安装教程:https://pjreddie.com/darknet/install/
遇到的坑
error darknet: ./src/cuda.c:36: check_error: Assertion `0' failed
根据下面网址修改MakeFile中ARCH的值
http://arnon.dk/matching-sm-architectures-arch-and-gencode-for-various-nvidia-cards/
编译完成之后
sudo rm -rf ~/.nv
至此,安装完成