记录一下ubuntu16.04安装NVIDIA显卡驱动+cuda9.0+cudnn7.0+opencv3.4+caffe+tensorflow+keras

第一步：安装NVIDIA显卡驱动

1.找机子的显卡

lspci | grep -i nvidia  

2.安装对应显卡的驱动，可以去官网查看，我的是1080Ti，对应 nvidia-384

sudo add-apt-repository ppa:graphics-drivers/ppa  
sudo apt-get update
sudo apt-get install nvidia-384

3.重启

第二步：安装cuda9.0

去官网查看，本文选择使用的本地 deb安装
https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb`
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
sudo apt-get update`
sudo apt-get install cuda`

添加cuda环境

 echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc  
 echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc  
  source ~/.bashrc

检查cuda版本

nvcc -v

重启，为啥要重启，防止循环登陆，解决办法另行解决

第三步：安装cudnn7.0

先去官网下载cudnn7.0，需要登陆才能下载,要登陆一下,然后下载一个cudnn-9.0-linux-x64-v7.5.0.56.solitairetheme8,这时候你们会奇怪,这是什么东西,这个其实就是压缩包,我们把它变成熟悉的tgz形式

 cp cudnn-9.0-linux-x64-v7.5.0.56.solitairetheme8 cudnn-9.0-linux-x64-v7.5.0.56.tgz
 tar -xvf cudnn-9.0-linux-x64-v7.5.0.56.tgz 
 sudo chmod -R 777 cuda
 cd cuda/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

第四步:安装tensorflow和keras

注意注意!一定要看tensorflow支持的版本,本文选择1.11.0

sudo pip3 install tensorflow-gpu==1.11.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

-i 后面的是加了一个清华镜像,不然安装特别慢,如果没有pip的需要装一个pip,安装keras

sudo pip3 install -U --pre keras -i https://pypi.tuna.tsinghua.edu.cn/simple

这边缺什么再装什么,比如

sudo pip3 install -U --pre numpy scipy matplotlib scikit-learn scikit-image -i https://pypi.tuna.tsinghua.edu.cn/simple

第五步:安装opencv3.4

参考网上教程,这里大致的写一下
1.安装相关包,缺什么补什么

sudo apt-get install cmake git pkg-config 
 sudo apt-get install libjpeg8-dev
 sudo apt-get install libtiff5-dev
sudo apt-get install libjasper-dev 
sudo apt-get install libpng12-dev
sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev
 sudo apt-get install libgtk2.0-dev
sudo apt-get install libatlas-base-dev gfortran
 sudo apt-get install qt5-default

mkdir build
cd build
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..
sudo make -j32
sudo make install 

有什么报错,解决什么报错
终端打开python
查看是能不能import python

第六步:安装caffe

安装相关包

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  
    sudo apt-get install python-skimage ipython python-pil python-h5py ipython python-gflags python-yaml  
    sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev  

    git clone https://github.com/BVLC/caffe.git  
    cd caffe  
    cp Makefile.config.example Makefile.config  

按照自己的配置对Make.config进行配置

    make all -j $(($(nproc) + 1))  
    make test -j $(($(nproc) + 1))  
    make runtest -j $(($(nproc) + 1))  
    make pycaffe -j $(($(nproc) + 1))  

以前总结过一个安装报错
ubuntu安装caffe以及遇到的问题

基本安装结构就是这样,当然中间会多次出现报错,需要一步一步解决