mxnet在windows使用gpu 出错

2021-02-04 15:17

阅读：357

标签：erro window ase imp jenkin use 运行 import cuda

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.]

mxnet在windows使用gpu 出错

标签：erro window ase imp jenkin use 运行 import cuda

原文地址：https://www.cnblogs.com/sunyongliang/p/11462545.html

上一篇：禁用Win10自带截图工具快捷键（Shift+Win+S）

下一篇：Acwing-99-激光炸弹(二维前缀和)

文章来自：搜素材网的编程语言模块，转载请注明文章出处。
文章标题：mxnet在windows使用gpu 出错
文章链接：http://soscw.com/index.php/essay/50952.html

亲，登录后才可以留言！

mxnet在windows使用gpu 出错

评论

热门文章

推荐文章

最新文章

置顶文章