ubuntu從零開始安裝mxnet--安裝mxnet

mxnet的安裝有多種方式，最簡單的自然是pip直接安裝。這裏只說明gpu版本。

pip安裝

python準備

安裝python, python-pip這些都不在贅述

安裝mxnet

pip install mxnet-cu80==0.11.0

測試mxnet

python
import mxnet as mx
a = mx.nd.ones((2, 3), mx.gpu())
b = a * 2 + 1
b.asnumpy()

若輸出結果如下

array([[ 3.,  3.,  3.],
       [ 3.,  3.,  3.]], dtype=float32)

表示安裝正常。

docker安裝

安裝docker

安照說明，我們逐步安裝
1. apt update
2. sudo apt-get install \
apt-transport-https \
ca-certificates \
curl \
software-properties-common
3. curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
4. sudo add-apt-repository \
"deb [arch=amd64] https://download.docker.com/linux/ubuntu \
$(lsb_release -cs) \
stable"
5. sudo apt-get update
6. sudo apt install docker-ce

安裝nvidia-docker-plugin

1. 於github中下載nvidia-docker-plugin的最新版本
2. 安裝nvidia-docker-plugin
   sudo dpkg -i nvidia-docker_1.0.1-1_amd64.deb
3. 安裝nvidia-modprobe, apt install nvidia-modprobe
4. 啓動nvidia-docker-plugin，nvidia-docker-plugin，如若正常表示安裝成功。

下載Mxnet Docker鏡像

1. sudo docker pull mxnet/python:gpu
2. 啓動鏡像nvidia-docker run -it mxnet/python:gpu bash
3. 和使用python安裝的方式一樣，我們測試執行上面的python，如果結果相同，表示安裝成功。

至此我們ubuntu從零開始安裝mxnet系列到此完結，感謝各位開發者的無私奉獻。