无法为此实例初始化gpu。FLOYDHUB CLI中出错:“GPU支持已禁用。使用\u CUDA=1编译MXNet以启用GPU支持。”

2024-09-28 05:24:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在FloydHub上训练我的MXNet YOLO对象检测模型,但是我不能为这个实例初始化GPU

ERROR IN FLOYDHUB CLI: "GPU support is disabled. Compile MXNet with USE_CUDA=1 to enable GPU support."

我使用pip安装mxnet-cu92mkl/cu92/cu100。试图用不同的CUDA版本加载我的工作,但每次都会出现相同的错误。 我正在使用FloydHub云平台

# training contexts
try:
    a = mx.nd.zeros((1,), ctx=mx.gpu(0))
    ctx = [mx.gpu(0)]
except:
    ctx = [mx.cpu()]
print(f"Training model on {ctx[0].device_type.upper()} ...")
print("Cpu or Gpu is decided decision=",ctx[0])

Tags: 对象实例模型supportgpuisyoloerror

热门问题