python -u predict.py --gpus 1 --img 1.jpg and python -u predict.py --gpus 0 --img 1.jpg-优快云博客

本文链接：https://blog.youkuaiyun.com/keyanxiaocaicai/article/details/52677990

本文记录了在使用predict.py脚本时遇到的错误及其解决过程。主要问题是由于将GPU设备列表错误地传递给了MXNet的NDArray函数导致的AttributeError异常。文章详细分析了错误原因并提供了修正方法。

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

python -u predict.py --gpus 1 --img 1.jpg

Traceback (most recent call last):

File "predict.py", line 100, in <module>
    main()
File "predict.py", line 61, in main
    arg_params['data'] = mx.nd.array(img, devs)
File "/home/ggj/ggj/pro/mxnet-face-master/mxnet/python/mxnet/ndarray.py", line 908, in array
    arr = empty(source_array.shape, ctx, dtype)
File "/home/ggj/ggj/pro/mxnet-face-master/mxnet/python/mxnet/ndarray.py", line 578, in empty
    return NDArray(handle=_new_alloc_handle(shape, ctx, False, dtype))
File "/home/ggj/ggj/pro/mxnet-face-master/mxnet/python/mxnet/ndarray.py", line 65, in _new_alloc_handle
    ctypes.c_int(ctx.device_typeid),
AttributeError: 'list' object has no attribute 'device_typeid'

*** Error in `python': realloc(): invalid old size: 0x00007fc790b18250 ***

python -u predict.py --gpus 0 --img 1.jpg

Traceback (most recent call last):
File "predict.py", line 100, in <module>
    main()
File "predict.py", line 61, in main
    arg_params['data'] = mx.nd.array(img, devs)
File "/home/ggj/ggj/pro/mxnet-face-master/mxnet/python/mxnet/ndarray.py", line 908, in array
    arr = empty(source_array.shape, ctx, dtype)
File "/home/ggj/ggj/pro/mxnet-face-master/mxnet/python/mxnet/ndarray.py", line 578, in empty
    return NDArray(handle=_new_alloc_handle(shape, ctx, False, dtype))
File "/home/ggj/ggj/pro/mxnet-face-master/mxnet/python/mxnet/ndarray.py", line 65, in _new_alloc_handle
    ctypes.c_int(ctx.device_typeid),
AttributeError: 'list' object has no attribute 'device_typeid'
*** Error in `python': free(): invalid pointer: 0x00007f13ae7cf7e0 ***
Aborted (core dumped)
这个问题：

是因为我想跑通toromo