在安装torchtext时用的pip install torchtext来安装,但这种安装方式会直接安装最新版的torchtext,并自动升级对应的torch版本导致gpu加速失效。
解决办法:在pytorch官网重新安装torch,并用pip install torchtext==具体版本号可以解决
pytorch对应的torchtext版本
import torch import torchtext a=torch.__version__ print(a) a=torch.cuda.is_available() print(a) a=torchtext.__version__ print(a)
查看具体版本以及GPU加速是否可用代码