ubuntu 16.04 安装MXNet GPU版本

安装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

安装脚本源代码

#!/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 -eMXNET_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 graphvizecho "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-numpyecho "Installing Python setuptools..."
sudo apt-get install -y python-setuptools python-pipecho "Installing Python package for MXNet..."
cd python; sudo python setup.py installecho "Adding MXNet path to your ~/.bashrc file"
echo "export PYTHONPATH=$MXNET_HOME/python:$PYTHONPATH" >> ~/.bashrc
source ~/.bashrcecho "Install Graphviz for plotting MXNet network graph..."
sudo pip install graphvizecho "Installing Jupyter notebook..."
sudo pip install jupyterecho "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

ubuntu 16.04 安装MXNet GPU版本相关推荐

ubuntu 16.04 安装TensorFlow GPU版本
推荐新版安装教程 http://blog.csdn.net/chenhaifeng2016/article/details/78874883 在ubuntu 16.04上安装cuda8.0和cudnn ...
ubuntu 16.04 安装Caffe GPU版本
安装nvidia显卡驱动和CUDA8.0和cuDNN5请参考以下内容 http://blog.csdn.net/chenhaifeng2016/article/details/68957732 安装第 ...
Ubuntu 20.04 安装Tensorflow GPU版本（GTX-1060显卡）
Ubuntu 20.04 安装Tensorflow GPU版本 (GTX-1060显卡) 参考: https://cyfeng.science/2020/05/02/ubuntu-install-nv ...
linux安装tensorflow教程,Ubuntu 16.04 安装 TensorFlow(GPU支持)
本文记录Ubuntu 16.04安装Tensorflow步骤,也包括怎么从源码编译安装Tensorflow. 要想安装Tensorflow GPU版本,你需要有一个新一点的Nvidia显卡. Tens ...
linux版本的qq怎么安装路径,Ubuntu 16.04安装QQ国际版图文详细教程
因工作需要,我安装了Ubuntu 16.04,但是工作上的很多事情需要QQ联系,然而在Ubuntu上的WebQQ很是不好用,于是在网上搜索了好多个Linux版本的QQ,然而不是功能不全,就是界面丑到爆 ...
Ubuntu 16.04 安装 caffe
Ubuntu16.04安装Caffe最全最详细教程(CPU) 由于本人电脑不支持GPU,故选择CPU安装caffe.具体查看方式:键入命令: lspci | grep -i nvidia ,若按回车之 ...
【转】Ubuntu 16.04 安装 CUDA10.1 （解决循环登陆的问题）
转自:Ubuntu 16.04 安装 CUDA10.1 (解决循环登陆的问题) - Hongkai_Ding - 博客园 0. 前言这里直接用 cuda安装文件同时安装 NVIDIA 驱动和 CUD ...
linux双系统重启卡光标,Ubuntu 16.04安装完重启后黑屏，光标一直闪
原文:https://blog.csdn.net/weixin_38533896/article/details/81023690 版权声明:本文为博主原创文章,转载请附上博文链接! 按照百度经验里面 ...
Ubuntu 16.04 安装 CUDA Toolkit 10.0
Ubuntu 16.04 安装 CUDA Toolkit 10.0 NVIDIA 引领人工智能计算 - NVIDIA https://www.nvidia.cn/ CUDA Zone https:// ...

ubuntu 16.04 安装MXNet GPU版本

ubuntu 16.04 安装MXNet GPU版本相关推荐

最新文章

热门文章