ubuntu从零开始安装mxnet--安装mxnet
mxnet的安装有多种方式,最简单的自然是pip直接安装。这里只说明gpu版本。pip安装python准备安装python, python-pip这些都不在赘述安装mxnetpip install mxnet-cu80==0.11.0测试mxnetpythonimport mxnet as mxa = mx.nd.ones((2, 3), mx.gpu())b = a * 2 + 1b.as
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
- 于github中下载nvidia-docker-plugin的最新版本
- 安装nvidia-docker-plugin
sudo dpkg -i nvidia-docker_1.0.1-1_amd64.deb
- 安装nvidia-modprobe,
apt install nvidia-modprobe
- 启动nvidia-docker-plugin,
nvidia-docker-plugin
,如若正常表示安装成功。
下载Mxnet Docker镜像
sudo docker pull mxnet/python:gpu
- 启动镜像
nvidia-docker run -it mxnet/python:gpu bash
- 和使用python安装的方式一样,我们测试执行上面的python,如果结果相同,表示安装成功。
至此我们ubuntu从零开始安装mxnet系列到此完结,感谢各位开发者的无私奉献。
更多推荐
所有评论(0)