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

保存后,创建链接文件

<
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值