ubuntu 16.04 安装MXNet GPU版本
2017-03-28 13:58
393 查看
安装MXNet for Ubuntu有两种方式。
方式一:安装预编译文件
pip install mxnet-cu80
方式二:编译源代码
安装nvidia显卡驱动和cuda/cudnn,请参考一下内容
http://blog.csdn.net/chenhaifeng2016/article/details/68957732
下载mxnet
cd /usr/local/src
git clone https://github.com/dmlc/mxnet.git --recursive
cd mxnet
cp make/config.mk .
vim config.mk
cd setup-utils
编辑install-mxnet-ubuntu-python.sh,由于默认不安装在$HOME/mxnet下面,所以需要更改MXNET_HOME
bash install-mxnet-ubuntu-python.sh
安装脚本源代码
运行example
cd example/image-classification
这里有个坑,运行测试代码前请先安装pip install requests
python train_mnist.py --network mlp
使用gpu加速
python train_mnist.py --network mlp --gpus 0
参考资料
http://mxnet.io/get_started/ubuntu_setup.html
方式一:安装预编译文件
pip install mxnet-cu80
方式二:编译源代码
安装nvidia显卡驱动和cuda/cudnn,请参考一下内容
http://blog.csdn.net/chenhaifeng2016/article/details/68957732
下载mxnet
cd /usr/local/src
git clone https://github.com/dmlc/mxnet.git --recursive
cd mxnet
cp make/config.mk .
vim config.mk
cd setup-utils
编辑install-mxnet-ubuntu-python.sh,由于默认不安装在$HOME/mxnet下面,所以需要更改MXNET_HOME
bash install-mxnet-ubuntu-python.sh
安装脚本源代码
#!/usr/bin/env bash ###################################################################### # This script installs MXNet for Python along with all required dependencies on a Ubuntu Machine. # Tested on Ubuntu 14.0 + distro. ###################################################################### set -e MXNET_HOME="/usr/local/src/mxnet/" echo "MXNet root folder: $MXNET_HOME" echo "Installing build-essential, libatlas-base-dev, libopencv-dev, pip, graphviz ..." sudo apt-get update sudo apt-get install -y build-essential libatlas-base-dev libopencv-dev graphviz echo "Building MXNet core. This can take few minutes..." cd "$MXNET_HOME" cp make/config.mk . make -j$(nproc) echo "Installing Numpy..." sudo apt-get install python-numpy echo "Installing Python setuptools..." sudo apt-get install -y python-setuptools python-pip echo "Installing Python package for MXNet..." cd python; sudo python setup.py install echo "Adding MXNet path to your ~/.bashrc file" echo "export PYTHONPATH=$MXNET_HOME/python:$PYTHONPATH" >> ~/.bashrc source ~/.bashrc echo "Install Graphviz for plotting MXNet network graph..." sudo pip install graphviz echo "Installing Jupyter notebook..." sudo pip install jupyter echo "Done! MXNet for Python installation is complete. Go ahead and explore MXNet with Python :-)"
运行example
cd example/image-classification
这里有个坑,运行测试代码前请先安装pip install requests
python train_mnist.py --network mlp
使用gpu加速
python train_mnist.py --network mlp --gpus 0
参考资料
http://mxnet.io/get_started/ubuntu_setup.html
相关文章推荐
- ubuntu16.04安装+cuda8.0+cudnn5.1+MXNET gpu版本安装+tensorflow gpu版本安装+chainerGPU版本安装
- ubuntu16.04 安装CUDA 8.0 和 cuDNN 5.1 /cudnn6.0,可适用于gpu版本的(tensorflow,caffe,mxnet)
- (亲测可用)ubuntu16.04安装+cuda8.0+cudnn5.1+MXNET gpu版本安装+tensorflow gpu版本安装+chainerGPU版本安装
- Ubuntu16.04+Anaconda 安装GPU版本tensorflow
- tensorflow:ubuntu16.04 gpu版本安装
- TensorFlow(gpu版本)安装教程 Ubuntu16.04
- Python 虚拟环境+安装MxNet(CPU和GPU版本)
- [置顶] Ubuntu16.04下安装Caffe-GPU版本的相关软件
- ubuntu 16.04 安装Caffe GPU版本
- Win10 64bit +VS2015+GPU版本MXNet+Tensorflow 1.3.0的安装
- ubuntu16.04 下安装加入opencv_contribute包的GPU版本的OpenCV3.2(CUDA支持)步骤与问题解决
- Ubuntu 16.04安装配置TensorFlow GPU版本
- 配有Tesla K40c的服务器新装Ubuntu16.04并安装CUDA8.0、Anaconda3、Matlab2016a、OPENCV3.1、CuDNN5.1、MXNet
- ubuntu 16.04 安装caffe GPU版本时遇到的问题
- Ubuntu16.04 安装配置GPU版本Caffe
- yolo-darknet在ubuntu下的GPU版本安装与测试
- 小米笔记本2 Ubuntu 16.04下安装Tensorflow GPU 1.2.1版本说明
- Ubuntu16.04下安装配置了tensorflow GPU版本后导致的常见错误
- ubuntu16.04 从源码安装tensorflow-gpu r1.4版本---2017-12-07
- ubuntu16.04 tensorflow-gpu版本安装好后,简单的检测代码