src/storage/storage.cc: Compile with USE_CUDA=1 to enable GPU usage【MXNET GPU版】

43 篇文章 2 订阅
在Win10环境下,运行MXNET时遇到GPU不支持的错误。原因是安装了CPU版MXNET,而非GPU版。解决方法是卸载现有CPU版本,根据CUDA版本(如cuda10.0对应mxnet-cu100)正确安装GPU版MXNET。安装错误的CUDA版本会导致运行时错误。
摘要由CSDN通过智能技术生成

【环境】Win10

一、想在GPU上运行MXNET,报错如下:

MXNetError: [23:23:48] src/storage/storage.cc:xxx: Compile with USE_CUDA=1 to enable GPU usage

【原因】安装的是cpu版的mxnet,不是gpu版的

【解决】卸载原来cpu版本的mxnet(如果猜的没错,你是通过pip install mxnet简单粗暴安装的),安装gpu版的mxnet 

安装哪个gpu版本的mxnet呢?看你电脑上装了哪个版本的cuda,(ubuntu在/usr/local/下可以看到)。如果是cuda10.0版,就安装mxnet-cu100(否则还会报错,报错内容见后文)。cuda9.0就是mxnet-cu90。其他的如cuda8.0可能就安装mxnet-cu80吧,没怎么试过。解决过程如下:

# 管理员模式进入cmd
C:\WINDOWS\system32>

# 卸载cpu版mxnet 
C:\WINDOWS\system32> pip uninstall mxnet

# 安装gpu版mxnet
C:\WINDOWS\s
### 回答1: MXNet错误:编译时使用use_cuda = 1以启用GPU使用。 这个错误意味着你需要在编译MXNet时使用use_cuda = 1选项来启用GPU使用。这通常需要安装CUDA和cuDNN,并确保它们与MXNet本兼容。如果您已经安装了CUDA和cuDNN,但仍然遇到此错误,请确保您的环境变量和路径设置正确。 ### 回答2: MXNetError: 编译时使用use_cuda = 1以启用GPU使用 MXNet是一个流行的深度学习框架,它提供了使用CPU和GPU来执行深度学习任务的能力。在许多情况下,使用GPU可以大大提高训练和推理的速度,因为GPU上的并行计算非常适合执行深度学习操作。当您在使用MXNet时遇到MXNetError: compile with use_cuda = 1 to enable GPU usage错误时,这意味着您的代码正在尝试使用GPU,但MXNet没有编译以支持GPU操作。 要解决此错误,您需要使用GPU编译MXNet。您可以按照以下步骤来完成: 1.安装CUDACUDA是NVIDIA提供的并行计算平台和编程模型,可用于加速深度学习任务。要使用MXNetGPU功能,您需要在计算机上安装CUDA。您可以从NVIDIA网站上下载与您的计算机配置相对应的CUDA本,并按照安装指南进行安装。 2.安装CuDNN:CuDNN是针对深度学习应用程序的CUDA库,它提供了高效的卷积神经网络和循环神经网络的实现。MXNet需要CuDNN来优化GPU上的计算。您可以从NVIDIA网站上下载与您的CUDA本相匹配的CuDNN本,并按照安装指南进行安装。 3.使用use_cuda = 1重新编译MXNet:一旦您安装了CUDA和CuDNN,您需要重新编译MXNet以使用GPU。您可以将use_cuda标志设置为1来启用GPU支持,并将其他MXNet选项设置为适当的值。在重新编译之前,您需要设置MXNetCUDA路径和CuDNN路径以便正确地连接CUDA和CuDNN库。重新编译MXNet可能需要一些时间,具体取决于您的计算机配置。 完成上述步骤后,您应该能够使用MXNetGPU功能,而不会再遇到MXNetError: compile with use_cuda = 1 to enable GPU usage错误。 总之,MXNetError: compile with use_cuda = 1 to enable GPU usage错误表明MXNet代码正在尝试使用GPU,但MXNet没有编译以支持GPU操作。要解决此错误,您需要安装CUDA和CuDNN,并重新编译MXNet以启用GPU支持。 ### 回答3: mxneterror: compile with use_cuda=1 to enable gpu usageMXNet深度学习框架在训练深度神经网络时会遇到的一个错误提示。这个错误提示的意思是说你需要在编译MXNet源代码时使用use_cuda=1的选项,才能启用GPU加速功能。 MXNet是一个支持跨平台的深度学习框架,它支持多种语言和运行环境,包括Python、R、C++等。在使用MXNet进行深度学习训练时,使用GPU可以大幅提升训练速度和效率。因此,如果你的MXNet程序需要使用GPU进行加速,那么你需要在编译MXNet源代码时加入use_cuda=1选项来启用GPU加速功能。 具体的操作步骤如下: 1. 下载MXNet源代码并解压缩到本地目录。 2. 安装CUDA和cuDNN库,这两个库是MXNet使用GPU加速的必备条件。 3. 打开命令行窗口,进入MXNet源代码目录,执行以下命令: make USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda USE_CUDNN=1 这个命令指定了MXNet编译时使用CUDA和cuDNN库,并启用了GPU加速功能。 4. 等待编译完成,获取MXNet的编译结果,使用这个编译结果替换原来的MXNet库文件。 如果编译成功,会在MXNet源代码目录下生成一个lib文件夹,这个文件夹下包含了编译好的MXNet库文件。 以上就是解决mxneterror: compile with use_cuda=1 to enable gpu usage的方法,希望能帮助你顺利完成MXNetGPU加速开发和训练。
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值