Caffe在Ubuntu 16.04的安装配置
安装配置网络搜索一大把,可是来回折腾好多回,记录整理过程,方便自己和他人:
- 独立显卡的Nvidia驱动安装
- Opencv安装
- 依赖库安装
- Caffe下载,配置与编译,测试
环境准备:安装编译依赖库
-协议、数据、opencv库等:sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
-Boost库 sudo apt-get install --no-install-recommends libboost-all-dev
- ATLAS库 sudo apt-get install libatlas-base-dev
(科学计算库BLAS(Basic Linear Algebra Subprograms 基本线性代数子程序))
- 日志、标签等库 sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
NVIDIA相关配置
(1)安装NVIDIA驱动
首先去官网(http://www.nvidia.com/Download/index.aspx?lang=en-us)查看适合自己显卡的驱动:
图1.显卡驱动查询
选择符合操作系统、显卡版本下载,本例采用run文件安装方式,如NVIDIA-Linux-x86_64-378.13.run.
因为是独显,需要重启ubuntu,高级模式里选recovery mode,选择network等选项后,选root …shell,进入命令行界面。
运行
sh ./NVIDIA-Linux-x86_64-378.13.run
然后重启系统。
终端执行命令 nvidia-smi ,若列出了GPU的信息列表则表示驱动安装成功。
(2)安装CUDA
CUDA是NVIDIA的编程语言平台,想使用GPU就必须要使用cuda。
首先下载CUDA
首先在官网上(https://developer.nvidia.com/cuda-downloads)下载CUDA:
接着安装CUDA
下载完成后执行以下命令:
sudo sh cuda_8.0.27_linux.