RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp cuda9.0+torch0.4 解决办法:在demo.py或者main.py加上 import torch torch.backends.cudnn.enabled = False