运行代码时候报错：

原因：pytorch与cuda版本不对，需要重新安装。不过我在复现代码的时候一般是要求特定的环境，不然会有其他错误，所以选择其他解决办法。

解决方案：

在train.py开头加入一行代码即可解决：

torch.backends.cudnn.enabled = False