【MXNet TX2】成功在TX2上编译MXNet源码,并使用GPU计算(踩坑与填坑)

本文详细记录了在TX2上成功编译MXNet源码并使用GPU计算的过程,包括遇到的“no kernel image”错误及其解决方案。关键在于修改`make/config.mk`,添加指定CUDA_ARCH的命令,确保与TX2的计算能力匹配。重新编译后,即可在TX2上进行MXNet的GPU开发。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

【MXNet TX2】成功在TX2上编译MXNet源码,并使用GPU计算(踩坑与填坑)

首先,便是去看官网的教程了,按照官网教程走,肯定可以成功编译mxnet,并可以执行python代码:

import mxnet as mx
mx.gpu()

但是,当你尝试执行gpu计算任务时,会提示 no kernel image **之类的错误。例如:

Check failed: (err) == (cudaSuccess
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

穆友航

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值