ubuntu 18.04 + caffe GPU安装/样例测试

ubuntu 18.04 + caffe GPU安装/样例测试

1. ubuntu 18.04安装

 折腾了2天centos+caffe，为了要桌面，选择了Gnome，但GPU显卡驱动装好后，init 5还能回到桌面，但reboot后，桌面就挂了，重装例几次系统，尝试例各种方式，最终都是一样但结果，没有桌面就意味着不能百度，就像黑夜中前行没有手电筒一样，最终放弃centos，转战ubuntu。
 安装系统需要先制作一个启动盘，推荐用U盘做启动盘，在windows上，上ubuntu官网下载最新但iso镜像，然后推荐用USBWriter（https://sourceforge.net/projects/usbwriter/files/latest/download）将iso制作成U盘启动盘，安装ubuntu很简单，一路默认就可以了。

2. GPU显卡驱动安装

  sudo apt --fix-broken install
  sudo ubuntu-drivers autoinstall

3. cuda/cudnn安装

  先到NVIDIA官网上下载对应版本的cuda和cudnn，我这里下载的是cuda_9.2.148_396.37_linux.run和cudnn-9.2-linux-x64-v7.2.1.38.tgz。
  然后安装cuda:
  sudo ./cuda_9.2.148_396.37_linux.run
  因为前面GPU显卡驱动已经安装了，这里出现询问是否安装GPU显卡加速驱动时，回答n。OpenGL是否安装时，也选择了n，因为很多人说最好不要装这个，centos但桌面挂怕了，所以没敢装OpenGL。其他都选择y，或默认就可以了。
  cudnn只用解压后拷贝到相应目录就可以了:
  tar -xzvf cudnn-9.2-linux-x64-v7.2.1.38.tgz
  sudo cp cuda/include/cudnn.h /usr/local/cuda-9.2/include
  sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.2/lib64
  gedit ~/.bashrc在末尾增加：
  export CUDA_HOME=/usr/local/cuda-9.2
  export PATH=/usr/local/cuda-9.2/bin:$PATH
  export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64:$LD_LIBRARY_PATH

4. caffe安装

  sudo apt install caffe-cuda

5. caffe样例测试

  经过上面的安装，caffe相关的命令都装到/usr/bin目录了，包括caffe, convert_imageset, convert_cifar_data, convert_mnist_data等等。
  在https://github.com/BVLC/caffe下载caffe-master.zip，并解压，此时已经不需要在编译caffe了，我们只需要用这里的examples。进入caffe-master目录，以cifar10为例，首先在data目录下载cifar10，然后
  vim examples/cifar10/create_cifar10.sh
  将convert_cifar_data.bin修改为convert_cifar_data并去掉前面的路径，因为现在这些命令都在/usr/bin目录了，可以直接使用。
  ./examples/cifar10/create_cifar10.sh
  vim examples/cifar10/train_quick.sh
  去掉caffe前面但路径。
  ./examples/cifar10/train_quick.sh
  经过5000次迭代，Test net output #0: accuracy = 0.7601
  Test net output #1: loss = 0.733249 (* 1 = 0.733249 loss)