在python中超简单安装mxnet

本文介绍了两种使用pip安装MXNet的方法,一种适用于CPU环境,另一种适用于GPU环境。对于CPU环境,直接运行pip install mxnet即可完成安装;对于GPU环境,则需要指定CUDA版本,如pip install mxnet-cu80。此外,还提供了安装后的测试代码。
摘要由CSDN通过智能技术生成

安装类别:

1) 用pip安装mxnet的python版本(CPU版本)
2) 用pip安装mxnet的python版本(GPU版本)
3) 编译安装到操作系统上

这里只说前两种,且不管是windows还是linux,python2还是python3,安装命令都一样

以下分别介绍

1, 用pip安装mxnet的python版本

C:\Users\someone>pip install mxnet
 100% |████████████████████████████████| 
C:\Users\someone>python
Python 3.5.2 |Continuum Analytics, Inc.| (default, Jul  5 2016, 11:41:13) [MSC v
.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import mxnet as mx
>>> a = mx.nd.ones((2,3))
>>> b  = a*2+1
>>> b

[[ 3.  3.  3.]
 [ 3.  3.  3.]]
<NDArray 2x3 @cpu(0)>

我不知道为什么这种方法这么简单,但是几乎没有看到有人用

2, 用pip安装mxnet的python版本(GPU版本)

C:\Users\someone>pip install mxnet-cu80
 100% |████████████████████████████████| 
C:\Users\someone>python
Python 3.5.2 |Continuum Analytics, Inc.| (default, Jul  5 2016, 11:41:13) [MSC v
.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import mxnet as mx
>>> a = mx.nd.ones((2, 3), mx.gpu())
>>> b  = a*2+1
>>> b

以上mxnet-cu80中cu80的意思是cuda8.0

评论 17
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值