在Google Colab使用transformers的Trainer时遇ImportError求助

错误提示 #

ImportError: Using the Trainer with PyTorch requires accelerate=0.20.1: Please run pip install transformers[torch] or pip install accelerate -U

解决方法 #

• 安装包含PyTorch依赖的完整transformers包：

  !pip install transformers[torch]
  

• 或单独升级accelerate库到指定版本以上：

  !pip install accelerate -U
  

涉及技术栈 #

transformers、GPT、PyTorch

内容的提问来源于stack exchange，提问作者jack wilson