Ubuntu16.04深度学习一键安装脚本(支持tensorflow、keras、caffe)
来源:互联网 发布:第四方支付钱包源码 编辑:程序博客网 时间:2024/05/01 18:23
引言
深度学习大热,因此很多同学有装机需求。本人在安装了许多台机器后,逐渐总结形成了一个“深度学习一键安装脚本”,可以在新装Ubuntu 16.04上一键安装 CUDA、cudnn、opencv、jupyter、深度学习库tensorflow、keras、caffe和Python编辑器 Pycharm。
脚本内容如下:
echo "One script installation for deep learning."echo "TomHeaven Presents."# configCAFFE_INSTALL=~echo "1 of 5: Install cuda and cudnn"# install nvidia driver and cudasudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.debsudo apt-get updatesudo apt-get install -y cuda# install cudnntar -xf cuda.tar.gzsudo cp cuda/include/* /usr/local/cuda/include/sudo cp cuda/lib64/* /usr/local/cuda/lib64/rm -r -f cudaecho "export LD_LIBRARY_PATH=\$LD_LIBRARY_PATH:/usr/local/cuda/lib64" | sudo tee -a /etc/profile# install pipsudo apt-get install -y python-pip# chmod for pip installation directoriessudo chmod a+rw -R /usr/lib/python2.7sudo chmod a+rw -R /usr/local/binsudo chmod a+rw -R /usr/local/sharesudo chmod a+rw -R /usr/local/lib/python2.7# upgrade pippip install --upgrade pipsudo cp /usr/bin/pip /usr/bin/pip.oldsudo apt-get remove -y python-pipsudo mv /usr/bin/pip.old /usr/bin/pipecho "2 of 5: Install opencv and others"# install opencvsudo apt-get install -y python-opencv libopencv*# otherssudo apt-get install -y openssh-server vim python-tk iptuxpip install scikit-learn pinyin jupyterecho "3 of 5: Install tensorflow and keras"# install tf, keraspip install tensorflow_gpu-1.2.0rc2-cp27-cp27mu-manylinux1_x86_64.whlpip install scipy-0.19.0-cp27-cp27mu-manylinux1_x86_64.whlpip install h5py keras scikit-image image# for visualizationsudo apt-get install -y graphvizpip install pydotecho "4 of 5: Install caffe"sudo apt-get install -y cmakesudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-devsudo 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 sudo apt-get install -y libgflags-dev libgoogle-glog-dev protobuf-compiler liblmdb-dev libatlas-base-dev doxygentar -zxvf caffe.tar.gzCAFFE_ROOT=$CAFFE_INSTALL/caffemv caffe $CAFFE_INSTALL/cd $CAFFE_ROOTmkdir buildcd buildcmake ..make -j $(($(nproc) + 1))make installecho "export PYTHONPATH=$CAFFE_ROOT/build/install/python:\$PYTHONPATH" | sudo tee -a /etc/profileecho "5 of 5: Install pycharm"# install pycharmtar -xf pycharm-community-2017.1.3.tar.gzrm -f ~/.local/share/applications/jetbrains-pycharm-ce.desktopsudo mv pycharm-community-2017.1.3 /opt/sudo ln -s -f /opt/pycharm-community-2017.1.3/bin/pycharm.sh /usr/bin/pycharm# run pycharmpycharm
用法
- 脚本依赖的包,本人都上传到了百度云。百度云下载地址:
链接:http://pan.baidu.com/s/1pKGV9OJ 密码:3tp9
- 下载所有内容到Ubuntu分区后,修改
install.sh
脚本的CAFFE_INSTALL=
为你需要的caffe安装路径,默认为当前用户的Home文件夹。 - 在
ubuntu_dl
目录打开终端,执行
chmod a+x install.sh./install.sh
然后输入密码即可。
注意事项
- 安装过程需要联网。
- 本脚本仅在新装
Ubuntu Kylin 16.04
系统测试无误。 - 脚本使用Ubuntu自带的Python 2.7,请不要安装Anaconda,可能会引发冲突。
- 安装完成后,在命令行中输入
source /etc/profilepythonimport tensorflowimport kerasimport 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。
阅读全文
0 0
- Ubuntu16.04深度学习一键安装脚本(支持tensorflow、keras、caffe)
- Ubuntu16.04+caffe+keras+tensorflow+GPU +matlab 安装资源查看
- 深度学习tensorflow安装ubuntu16.04amd64
- ubuntu16.04 64位 cpu安装tensorflow+theano+keras+caffe+xgboost
- 深度学习环境搭建 ubuntu16+GTX1080+CUDA8+Theano+Tensorflow+keras
- 深度学习之安装 TensorFlow Keras
- Ubuntu16.04+GTX1050ti+CUDA8.0+TensorFlow-gpu+Keras+Pycharm配置深度学习环境
- 深度学习框架搭建 Ubuntu16.04+CUDA+Anaconda4.2+Python3.5+keras+TensorFlow gpu+cuDNN
- Ubuntu16.04+CUDA+CUDNN+Anaconda+Tensorflow+keras深度学习环境搭建
- 深度学习库大排名:TensorFlow、Keras、caffe稳居前三
- Ubuntu16.04安装CUDA7.5+Caffe+tensorflow
- ubuntu16.04 下caffe+tensorflow+GPU 安装
- 深度学习(四十一)cuda8.0+ubuntu16.04+theano、caffe、tensorflow环境搭建
- 深度学习(四十一)cuda8.0+ubuntu16.04+theano、caffe、tensorflow环境搭建
- 深度学习服务器环境搭建详细版(Ubuntu16.04+CUDA8+Caffe+Anaconda+TensorFlow+共享)
- 深度学习框架keras安装(后端基于Tensorflow/theano)
- 【深度学习】Anaconda+Tensorflow+Keras Win7 64位安装i
- 为深度学习环境安装开源深度学习框架Tensorflow(Ubuntu16.04+CUDA+cuDNN+Anaconda+Tensorflow)
- LSTM理论学习
- 我的第一片博客
- git提交顺序
- (系统)商店购物管理系统
- Lesson 11 算术运算符与算术表达式 实践项目:1-分离各位数 2-分离整数和小数部分 项目3-如何买玫瑰? 项目4-玩数字 项目5-坐标转换
- Ubuntu16.04深度学习一键安装脚本(支持tensorflow、keras、caffe)
- 排序算法知识点
- 3541414.html
- 2017年迄今最新人工智能资源盘点, 分享给你, 快来看看吧!
- HorizontalScrollView横向滑动
- 双边滤波加速
- java实现TreeSet,迭代器使用二叉查找树,每个节点有父节点链
- 模式总结
- 计算机网络知识点总结