Bug解決： invalid argument at /pytorch/aten/src /THC/THCGeneral.cpp:405

原文鏈接：https://discuss.pytorch.org/t/a-error-when-using-gpu/32761

錯誤信息：

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，此時雖然報錯但並不影響後面的運行。