下载安装Torch GPU 加速版

首先确定电脑是否支持gpu加速

1 找到控制面板，切换大图标 然后NVIDIA控制面板，废话不多说，直接上图

安装pythorch

方法一
官网下载地址：https://pytorch.org/get-started/locally/

注意这里我们选择安装CUDA对应的版本，为了安装过程顺利，选择版本相符合的最好。

pip命令
pip install torch=1.7.0 torchvision=0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

方法二
下载.whl文件，网址：https://pytorch.org/get-started/previous-versions/
win+r输入cmd回车

cd (下载目录)
pip install torch-1.6.0-cp37-cp37m-win_amd64.whl
pip install transformers

验证是否成功

python
import torch
torch.__version__
torch.cuda.is_available()