一、 英伟达驱动安装
sudo apt-get update
英伟达显卡驱动向下兼容,所以只需要安装最新驱动即可。
运行:
ubuntu-drivers devices
显示推荐的英伟达驱动
运行apt安装驱动即可完成驱动安装:
sudo apt-get install nvidia-***
二、 CUDA安装
进入下载的文件路径
sudo sh cuda_9.2.88_396.26_linux.run
出现说明之后,按Ctrl + C键跳过,接受之后会问你是否需要安装驱动,选择无,其余选择是直至安装完成。
配置环境变量,运行如下命令打开个人资料文件
sudo gedit /etc/profile
打开文件后在文件末尾添加路径,也就是安装目录,命令如下:
export PATH=/usr/local/cuda-9.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64$LD_LIBRARY_PATH
保存,然后重启电脑,进入终端
$ cd /usr/local/cuda/samples/1_Utilities/deviceQuery #本人安装为CUDA10.1版本
$ sudo make
$ ./deviceQuery
如果result = PASS,则安装成功,否则失败,可能原因是环境未配置好,或者是cuda与驱动版本不兼容。
如果需要卸载重装,运行下列命令卸载 cuda
$ cd /usr/local/cuda/bin
$ sudo ./uninstall_cuda_9.2.pl
三、 安装opencv
sudo apt install libopencv-dev
四、安装darknet
将文件夹下的安装包复制到home目录。
Cd到darknet目录。
更改makefile前几行:
GPU=1
OPENCV=1
运行make命令。
安装完成后在控制台输入:
./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg
验证安装正确。
五、迁移学习