在pycharm中运行深度学习训练模型的代码，之前运行过并没报错，今天突然出了幺蛾子。为了防止广大网友像我这个倒霉蛋一样在网上找不到任何线索，在此提供一些小建议。
“got an unexpected keyword argument 'dropout'”这个错误发生在使用PyTorch版本低于1.7.0的情况下，因为在较早的版本中，MultiheadAttention类没有名为dropout的参数。
即使此时查看自己下载好pytorch依赖包的版本满足大于1.7.0的条件，也有可能出现这样的报错。解决办法是去官网重新下载一次：
根据具体版本需求、电脑操作系统、包、代码语言以及电脑配置是CPU还是CUDA以及CUDA版本选择，然后复制Run this Command框内容，在终端直接运行即可。
通常使用pip下载速度比Conda的更快。
如果不知道自己电脑有无CUDA或者想查看CUDA版本，快捷键Ctrl+r，输入cmd
在弹出的prompt框输入nvdia-smi，即可查看CUDA版本，例如我的电脑是12.2。
希望能帮到大家：）
扫一扫
2万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
