一、安装 PyTorch(cpu)
1. 访问 PyTorch 官方网站:PyTorch 。在INSTALL PYTORCH模块选择要安装的版本。找到上一步CUDA版本支持的PyTorch版本
命令如下:
pip3 install torch torchvision torchaudio
三、安装 Transformers
在命令行中直接使用pip安装
pip install transformers
一、安装 PyTorch(cpu)
1. 访问 PyTorch 官方网站:PyTorch 。在INSTALL PYTORCH模块选择要安装的版本。找到上一步CUDA版本支持的PyTorch版本
命令如下:
pip3 install torch torchvision torchaudio
三、安装 Transformers
在命令行中直接使用pip安装
pip install transformers