错误信息:

Starting preparing ...
GPU0 used: 92.0M
Traceback (most recent call last):
  File "main.py", line 109, in <module>
    main()
  File "main.py", line 53, in main
    p = Processor(args)
  File "I:\RA-GCNv1-master\src\processor.py", line 28, in __init__
    self.device_type = U.check_gpu(args.gpus)
  File "I:\RA-GCNv1-master\src\utils.py", line 11, in check_gpu
    handle = pynvml.nvmlDeviceGetHandleByIndex(i)
  File "D:\Anaconda3\envs\RAGCN\lib\site-packages\pynvml.py", line 946, in nvmlDeviceGetHandleByIndex
    _nvmlCheckReturn(ret)
  File "D:\Anaconda3\envs\RAGCN\lib\site-packages\pynvml.py", line 405, in _nvmlCheckReturn
    raise NVMLError(ret)
pynvml.NVMLError_InvalidArgument: Invalid Argument

 

gpu启动报错,把

pynvml.nvmlInit()
        for i in gpus:
            handle = pynvml.nvmlDeviceGetHandleByIndex(i)

改为pynvml.nvmlInit()
        for i in gpus:
            handle = pynvml.nvmlDeviceGetHandleByIndex(0)

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐