注意:你需要根据你的CUDA版本选择合适的PyTorch版本。例如,如果你的CUDA版本是11.1,则应选择支持CUDA 11.1的PyTorch版本。 验证安装是否成功: 你可以通过以下代码来验证reformer_pytorch和PyTorch是否正确安装,并且可以在你的环境中使用:python import torch from reformer_pytorch import ReformerLM # 检查PyTorch是否支持CUD...
安装reformer_pytorch 需要先pip install reformer_pytorch 然后执行第一步,安装torch-gpu版本