Mxnet框架搭建

Mxnet是亚马逊开发的深度学习框架,和谷歌Tensorflow是同类型的框架。

1.安装Mxnet

这里只展示在线安装,源码编译安装等不演示;GPU安装与Mxnet无关,只需安装不同的Mxnet版本就可以。要使用GPU还是得另外配置好Cuda和Cudnn。

(1)安装依赖

sudo apt-get update
sudo apt-get install -y wget python gcc
wget https://bootstrap.pypa.io/get-pip.py && sudo python get-pip.py

(2)安装Mxnet

pip install mxnet

注:要安装GPU版本,命令是pip install mxnet-cu90,根据Cuda版本,安装不同mxnet-cuxx版本,具体查官网信息

(3)安装Graphviz(非必须)

sudo apt-get install graphviz
pip install graphviz

2.测试是否安装完成

(1)进入python命令编辑界面

python

(2)依次每行拷贝以下命令进行测试,最后输入一致,即表示安装完成。

>>> import mxnet as mx
>>> a = mx.nd.ones((2, 3))
>>> b = a * 2 + 1
>>> b.asnumpy()
array([[ 3.,  3.,  3.],
       [ 3.,  3.,  3.]], dtype=float32)

(3)退出python控制台

exit()

参考资料:

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容