Mxnet框架搭建

Mxnet框架搭建

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

1.安装Mxnet

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

(1)安装依赖

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

(2)安装Mxnet

  1. pip install mxnet 

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

(3)安装Graphviz(非必须)

  1. sudo apt-get install graphviz 
  2. pip install graphviz 

2.测试是否安装完成

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

  1. python 

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

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

(3)退出python控制台

  1. exit() 

参考资料:

转载于:https://www.cnblogs.com/wushaogui/p/8782325.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值