我试图在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])
目前没有回答
相关问题 更多 >
编程相关推荐