mxnet在windows使用gpu 出錯

1. cuda windows安裝 官網下載

代碼：

import mxnet as mx
from mxnet import nd
from mxnet.gluon import nn


a = nd.array([1, 2, 3], ctx=mx.gpu(0))
print(a)

異常：mxnet.base.MXNetError: [23:42:39] C:\Jenkins\workspace\mxnet-tag\mxnet\src\storage\storage.cc:143: Compile with USE_CUDA=1 to enable GPU usage

原因：默認安裝mxnet是cpu版本，卸載重裝gpub版本 mxnet-cu91，和下載的cuda版本一致。

運行結果：

[1. 2. 3.]
<NDArray 3 @gpu(0)>