ubuntu16.04下一套深度学习工具的安装
第一步:安装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以及遇到的问题
基本安装结构就是这样,当然中间会多次出现报错,需要一步一步解决