Ubuntu16.04深度学习一键安装脚本(支持pytorch、tensorflow、keras、caffe)

前端之家收集整理的这篇文章主要介绍了Ubuntu16.04深度学习一键安装脚本(支持pytorch、tensorflow、keras、caffe)前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

引言

深度学习大热,因此很多同学有装机需求。本人在安装了许多台机器后,逐渐总结形成了一个“深度学习一键安装脚本”,可以在新装Ubuntu 16.04上一键安装 CUDA、cudnn、opencv、jupyter、深度学习库pytorch、tensorflow、keras、caffe和Python编辑器 Pycharm。

脚本内容如下:

  1. echo "One script installation for deep learning."
  2. echo "TomHeaven Presents @ 2018.03.13."
  3. # config
  4. CAFFE_INSTALL=~
  5.  
  6. echo "1 of 6: Install cuda and cudnn"
  7. # install nvidia driver and cuda
  8. sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
  9. sudo apt-get update
  10. sudo apt-get install -y cuda
  11.  
  12. # install cudnn
  13. tar -xf cudnn-8.0-linux-x64-v6.0.tgz
  14. sudo cp cuda/include/* /usr/local/cuda/include/
  15. sudo cp cuda/lib64/* /usr/local/cuda/lib64/
  16. rm -r -f cuda
  17. echo "export LD_LIBRARY_PATH=\$LD_LIBRARY_PATH:/usr/local/cuda/lib64" | sudo tee -a /etc/profile
  18.  
  19.  
  20. # install pip
  21. sudo apt-get install -y python-pip
  22. # chmod for pip installation directories
  23. sudo chmod a+rw -R /usr/lib/python2.7
  24. sudo chmod a+rw -R /usr/local/bin
  25. sudo chmod a+rw -R /usr/local/share
  26. sudo chmod a+rw -R /usr/local/lib/python2.7
  27.  
  28. # change pip source to Aliyun
  29. mkdir ~/.pip
  30. cp pip.conf ~/.pip/
  31.  
  32. # upgrade pip
  33. pip install --upgrade pip
  34. sudo cp /usr/bin/pip /usr/bin/pip.old
  35. sudo apt-get remove -y python-pip
  36. sudo mv /usr/bin/pip.old /usr/bin/pip
  37.  
  38.  
  39. echo "2 of 6: Install opencv and others"
  40. # install opencv
  41. sudo apt-get install -y python-opencv libopencv*
  42. # others
  43. sudo apt-get install -y openssh-server vim python-tk iptux
  44. pip install scikit-learn pinyin
  45. sudo pip install jupyter
  46.  
  47. echo "3 of 6: Install tensorflow and keras"
  48. # install tf,keras
  49. sudo pip install tensorflow_gpu-1.4.0-cp27-cp27mu-manylinux1_x86_64.whl
  50. pip install scipy
  51. pip install h5py keras
  52. # sudo pip install image
  53.  
  54. # for visualization
  55. sudo apt-get install -y graphviz
  56. pip install pydot
  57.  
  58. echo "4 of 6: Install pytorch"
  59. pip install torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl
  60. pip install torchvision
  61.  
  62.  
  63. echo "5 of 6: Install caffe"
  64. sudo apt-get install -y cmake
  65. sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev
  66. sudo apt-get install -y libboost-system-dev libboost-filesystem-dev libboost-thread-dev libboost-chrono-dev libboost-date-time-dev libboost-atomic-dev libboost-python-dev
  67. sudo apt-get install -y libgflags-dev libgoogle-glog-dev protobuf-compiler liblmdb-dev libatlas-base-dev doxygen
  68.  
  69. sudo pip install scikit-image
  70.  
  71. old_dir=`pwd`
  72.  
  73. tar -zxvf caffe.tar.gz
  74. CAFFE_ROOT=$CAFFE_INSTALL/caffe
  75. mv caffe $CAFFE_INSTALL/
  76. cd $CAFFE_ROOT
  77. mkdir build
  78. cd build
  79. cmake ..
  80. #make -j $(($(nproc) + 1))
  81. make -j 4
  82. make install
  83. echo "export PYTHONPATH=$CAFFE_ROOT/build/install/python:\$PYTHONPATH" | sudo tee -a /etc/profile
  84.  
  85. cd $old_dir
  86.  
  87.  
  88. echo "6 of 6: Install pycharm"
  89. # install pycharm
  90. tar -xf "pycharm-community-2017.3.3.tar.gz"
  91. rm -f ~/.local/share/applications/jetbrains-pycharm-ce.desktop
  92. sudo mv pycharm-community-2017.3.3 /opt/
  93. sudo ln -s -f /opt/pycharm-community-2017.3.3/bin/pycharm.sh /usr/bin/pycharm
  94.  
  95. # run pycharm
  96. pycharm

用法

  1. 链接:http://pan.baidu.com/s/1pKGV9OJ 密码:3tp9
  • 下载所有内容到Ubuntu分区后,修改install.sh脚本的CAFFE_INSTALL=为你需要的caffe安装路径,默认为当前用户的Home文件夹。
  • ubuntu_dl目录打开终端,执行
  1. chmod a+x install.sh
  2. ./install.sh

然后输入密码即可。

注意事项

  • 安装过程需要联网。
  • 本脚本仅在新装Ubuntu Kylin 16.04系统测试无误。
  • 脚本使用Ubuntu自带的Python 2.7,请不要安装Anaconda,可能会引发冲突。
  • 安装完成后,在命令行中输入
  1. source /etc/profile
  2. python
  3. import tensorflow
  4. import keras
  5. import caffe

应当无错。注意对/etc/profile修改重启系统后才会全局生效,因此在导入caffe之前需要手动执行它。重启后就无需再手动执行它了。

  • [2017.09.09更新] 华硕(Asus)主板的机器安装前请先看 Ubuntu16.04安装时常见故障及其解决方法中的故障四的解决方法。否则重启系统后有可能遇到循环登录问题。
  • [2017.09.09更新] Tensorflow和CuDNN的升级方法:本文提供的是Tensorflow 1.2和CuDNN5。如果需要Tensorflow 1.3,将脚本第3部分安装Tensorflow的语句改为pip install tensorflow-gpu,另外将还需要cuda.tar.gz文件替换为从Nvidia官网下载的CuDNN6。
  • [2018.03.13更新] 添加了Pytorch支持修改pip源为阿里云,大幅加速pip包下载速度;修复了由于软件源更新导致某些自动执行权限不足的问题。当日测试可以完美一键执行。

猜你在找的Ubuntu相关文章