ubuntu16.04 caffe cuda9.1 segnet nvidia gpu安装注意的点

GPU驱动：R390

cuda：9.1

gcc：5.4.0

anaconda：2

GPU运算能力：2.1

CPU：8G

系统：ubuntu 16.04 x86_64

 

安装一般依赖项：

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev

 还有官网提到的：

• BLAS via ATLAS, MKL, or OpenBLAS.
• Boost >= 1.55
• protobuf, glog, gflags, hdf5

以及anaconda，这些没什么特别注意的，我选择的是anaconda2,因为我要用segnet，不过我也配了3,版本可以说是文件夹，库都在那放着，你要谁，你就在caffe的config中指定谁，他就会去找，所以理论上来讲，你无比自由想用谁用谁，你可以一个项目配一个caffe。我就是这么干的。甚至不同的数据库还称一个caffe。一个不爽想从gpu换cpu那就直接make clean，然后你的caffe大楼就会井然有序的拆除。暂时还没出现什么问题。等我看看caffe 官方doc

安装驱动

1.很重要的是：如果你用ubuntu那你的显卡驱动以及cuda都必须！用apt安装！，要不cuda可能会崩溃，原文：

‘for CUDA version. Note, the cuda version may break if your NVIDIA driver and CUDA toolkit are not installed by APT.’

所以推荐：http://www.cnblogs.com/xujianqing/p/6142963.html这篇文中的显卡驱动安装方式，但是！！！版本号要从375换到390，因为cuda9要求驱动390

安装cuda

官网要求的gcc版本是5.3.1，然而自带的是5.4,不过我依旧安装成功了。如果有问题再看吧。而且在http://caffe.berkeleyvision.org/install_apt.html caffe官网的ubuntu安装页，写了16.04要用版本8的cuda，然而网上有不少9的教程，于是我来试试小白鼠好了。而且我的gpu运算能力只有2.1。可以看出我这是在作死的边缘试探。

 

 

https://developer.nvidia.com/cuda-gpus包括可以安装的CUDA的Gpu型号

1.首先你要看https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#package-manager-metas，里面有个kernel版本安装要求，这个版本必须匹配你电脑的版本，而且在cuda之前安装，因为如果你跳过这个步骤，那么cuda driver就会自己找个版本。

具体命令

sudo apt-get install linux-headers-$(uname -r)

 

2.安装cuda现在官网提供的最新版本9.1，要求R390驱动，如果你想要这个版本的cuda，直接复制粘贴网上的命令行，我也不知道会出现什么。

‘Before installing the CUDA Toolkit on Linux, please ensure that you have the latest NVIDIA driver R390 installed. The latest NVIDIA R390 driver is available at: www.nvidia.com/drivers’

并且推荐使用官网提供的命令行，以及终端建议的命令行。

 

3.安装library package工具包

参见https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#package-manager-metas   

3.7节，我就不翻译了命令行是：

cat /var/lib/apt/lists/*cuda*Packages | grep "Package:"

sudo apt-get install cuda 

sudo apt-get install cuda-drivers  
测试安装cuda

cd /usr/local/cuda-9.1/samples/1_Utilities/deviceQuery

make

sudo ./deviceQuery