错误信息:
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)