当前位置:   article > 正文

Caffe在Ubuntu 16.04的安装配置

caffe在ubuntu 16.04

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.ru

本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号