如何使用 cuda >=11.0（和 pytorch 1.9）安装 torchtext？

Question

我尝试过多种方法，例如

conda install -y pytorch==1.9 torchvision torchaudio torchtext cudatoolkit=11.0 -c pytorch -c nvidia

但由于某种原因，它永远不会下载 cuda 11.0 或更高版本的版本。

错误消息太大而无法粘贴，但您可以在此处查看详细信息：https://github.com/pytorch/text/issues/1395

使用空环境应该很容易重现，如下所示：

conda create -n env_a40 python=3.9
conda activate env_a40
conda install -y pytorch==1.9 torchvision torchaudio torchtext cudatoolkit=11.0 -c pytorch -c nvidia

交叉发布：

• https://discuss.pytorch.org/t/how-does-one-install-a-torchtext-version-兼容-with-cuda-11-0/132276
• https://github.com/pytorch/text/issues/1395

有关的：

• 如何在一台似乎拒绝合作的 HPC 中安装 pytorch 1.9？
• https://github.com/pytorch/text/issues/1397

---

请注意，您也可以使用 pip 尝试：

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

还没有成功。

Answer 1

根据文档页面，PyTorch 的编译版本 1.9 仅适用于平台CUDA 10.2和CUDA 11.1。

看来您确实正在使用CUDA 11.1（来自您对在您的环境中运行的评论module load cuda-toolkit/11.1）。因此，您应该安装 PyTorch 版本1.9.0-py3.9_cuda11.1_cudnn8.0.5_0，即使用以下命令：

conda install -y pytorch==1.9 torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

如果您使用 pip，请执行以下操作：

pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install torchtext==0.10.1

可能可以压缩为：

pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 torchtext==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

但还没有尝试过。