Pytorch——安装torchtext-cuda

Pytorch与torchtext的版本对应关系

安装Pytorch和torchtext

• 我的cuda版本为11.1，因此用以下命令安装：

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

• 安装对应版本torchtext：

pip install torchtext==0.10.0

即可成功安装~

参考文章：

• 记录安装torchtext会自动更新pytorch版本导致gpu加速失效问题