Mxnet框架的环境搭建
Mxnet是亚马逊开发的深度学习框架,和谷歌Tensorflow是同类型的框架。
1.安装Mxnet
这里只展示在线安装,源码编译安装等不演示;GPU安装与Mxnet无关,只需安装不同的Mxnet版本就可以。要使用GPU还是得另外配置好Cuda和Cudnn。
(1)安装依赖
1 | sudo apt-get update |
(2)安装Mxnet
1 | pip install mxnet |
注:要安装GPU版本,命令是pip install mxnet-cu90
,根据Cuda版本,安装不同mxnet-cuxx版本,具体查官网信息
(3)安装Graphviz(非必须)
1 | sudo apt-get install graphviz |
2.测试是否安装完成
(1)进入python命令编辑界面
1 | python |
(2)依次每行拷贝以下命令进行测试,最后输入一致,即表示安装完成。
1 | >>> import mxnet as mx |
(3)退出python控制台
1 | exit() |
参考资料: