Faster R-CNN python版在Ubuntu16.04环境下配置编译
1 电脑配置
在windows上进行配置了好几天,主要的caffe编译不成功导致失败,只好转战ubuntu。ubuntu安装建议选择16.04.5版本,会省好多事,安装过程就不说了,大家都会。先看一下电脑配置
CPU:Intel(R) Core(TM) i5-8300H CPU @ 2.30GHz四核
显卡:Nvidia GeForce 1050Ti
操作系统:Ubuntu 16.04
2 安装显卡驱动
安装显卡驱动有两种方法,一个是直接去软件与更新里选择Additional Drivers,直接选择安装。
另一种是在终端安装,打开终端,加入官方ppa源:
sudo add-apt-repository ppa:graphics-drivers/ppa
然后更新软件库安装驱动,命令行输入:
sudo apt-get update
sudo apt-get install nvidia-384 nvidia-settings nvidia-prime
384时版本号,安装后在终端输入 nvidia-smi 可以看到显卡信息。
3 安装CUDA
3.1 安装CUDA
首先去官网下载CUDA8.0的run文件(不要尝鲜去下9或者10,不然出问题都没地去找)。
下载完成之后,cd进入文件所在目录,获得权限后进行安装。
chmod 777 cuda_7.5.18_linux.run
sudo ./cuda_7.5.18_linux.run --override
注意后面的override是必须的,这样才能保证安装的过程中,不会出现编译器不支持的错误。另外,在选择条件的过程中,一定不要再次安装nvidia驱动,虽然cuda.run文件本身是包含又nvidia驱动的,但是本处直接安装会出错。
1). 启动安装程序,一直按空格到最后,输入accept接受条款
2). 输入n不安装nvidia图像驱动,之前已经安装过了
3). 输入y安装cuda 8.0工具
4). 回车确认cuda默认安装路径:/usr/local/cuda-8.0
5). 输入y用sudo权限运行安装,输入密码
6). 输入y或者n安装或者不安装指向/usr/local/cuda的符号链接
7). 输入y安装CUDA 8.0 Samples,以便后面测试
8). 回车确认CUDA 8.0 Samples默认安装路径:/home/yuan(yuan是我的用户名),该安装路径测试完可以删除
3.2 安装CUDNN
然后安装对应的CUDNN,去网站 https://developer.nvidia.com/rdp/cudnn-download 下载对应的CUDNN,需要注册。
然后进入文件所在目录,解压,并将头文件,动态文件进行复制和链接。
tar zxvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件
sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* #给所有用户增加这些文件的读权限
sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件
sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so
设置环境变量,终端输入
sudo gedit /etc/profile
在末位加上
PATH=/usr/local/cuda/bin:$PATH
export PATH
保存后,创建链接文件
<