资源分类
》 资源 》 经验分享 》mxnet在windows使用gpu 出错

mxnet在windows使用gpu 出错

2019-11-23 | |  83 |   0

原标题：mxnet在windows使用gpu 出错

原文来自：博客园      原文链接：https://www.cnblogs.com/sunyongliang/p/11462545.html


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)>


免责声明：本文来自互联网新闻客户端自媒体，不代表本网的观点和立场。

合作及投稿邮箱：E-mail:editor@tusaishared.com

上一篇：手把手教你用深度学习做物体检测(六)：YOLOv2介绍

下一篇：用深度学习做命名实体识别(二)：文本标注工具brat

用户评价
全部评价

热门资源

  • Python 爬虫（二）...

    所谓爬虫就是模拟客户端发送网络请求，获取网络响...

  • TensorFlow从1到2...

    原文第四篇中，我们介绍了官方的入门案例MNIST，功...

  • TensorFlow从1到2...

    “回归”这个词，既是Regression算法的名称，也代表...

  • 机器学习中的熵、...

    熵 (entropy) 这一词最初来源于热力学。1948年，克...

  • TensorFlow2.0（10...

    前面的博客中我们说过，在加载数据和预处理数据时...