mxnet在windows使用gpu 出错

Posted sunyongliang

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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)>

以上是关于mxnet在windows使用gpu 出错的主要内容,如果未能解决你的问题,请参考以下文章

Win10 + Python + GPU版MXNet + VS2015 + RTools + R配置

在 Windows 环境中,如果 GPU 已经使用,Cupy 在 multithread.pool 中会出错,即使我使用了 multiprocessing.set_start_method('spaw

动手学深度学习Jupyter notebook中 import mxnet出错

拥抱MXNet深度学习第二天(继续纠结安装环境)

『MXNet』专题汇总

MXNet GluonTS使用报错:OSError: libnccl.so.2: cannot open shared object file: No such file or directory