conda正确安装mxnet

错误安装:

pip install mxnet -i https://pypi.tuna.tsinghua.edu.cn/simple

报错

terminate called after throwing an instance of 'dmlc::Error'
  what():  [13:18:13] src/storage/storage.cc:119: Compile with USE_CUDA=1 to enable GPU usage
Stack trace:
  [bt] (0) ~/.conda/envs/head_detection_lffd/lib/python3.6/site-packages/mxnet/libmxnet.so(+0x307d3b) [0x7f90453f6d3b]
  [bt] (1) ~/.conda/envs/head_detection_lffd/lib/python3.6/site-packages/mxnet/libmxnet.so(+0x3c20a35) [0x7f9048d0fa35]

错误原因,需要安装GPU版本mxnet

正确安装

查看cuda版本
cat /usr/local/cuda/version.txt
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
安装gpu版本mxnet
pip install mxnet-cu100 -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值