1. cuda windows安装 官网下载

代码:

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。
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)>

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄