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:Jenkinsworkspacemxnet-tagmxnetsrcstoragestorage.cc:143: Compile with USE_CUDA=1 to enable GPU usage
原因：默认安装mxnet是cpu版本，卸载重装gpub版本 mxnet-cu91，和下载的cuda版本一致。
运行结果：
[1. 2. 3.]
<NDArray 3 @gpu(0)>
内容来源于网络如有侵权请私信删除
- 还没有人评论，欢迎说说您的想法！
客服
