CUDA bug解决

错误信息：

RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:405

参考：
https://discuss.pytorch.org/t/a-error-when-using-gpu/32761

原因：

• 多GPU测试
• PyTorch版本与显卡不兼容

在这里我的报错是由于后者，我的显卡是 RTX2080(Ti)，PyTorch 1.0，如果换成 RTX1080(Ti)就没有问题，比较简单的解决方案是，将 python 文件中的 torch.backends.cudnn.benchmark = True 设置成 False，即可得到一个静态 CUDA error，此时虽然报错但并不影响后面的运行。