如何使用 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://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

但没试过

Answer 2

对我来说，它适用于 torchtext 0.10.1。我做事的顺序是先安装 pytorch：

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

（可能使用pytorch网站上的最新命令https://pytorch.org/get-started/locally/, but if that doesn't work then go to the torchtext website to see what versions of python and pytorch they support and install that. Hopefully in the future torchtext will be in line with the main pytorch branch https://github.com/pytorch/text）

然后因为我使用的是我的个人库，所以我将其安装为可编辑模式：

pip install -e ~/ultimate-utils/ultimate-utils-proj-src

或来自 pypi

pip install ultimate-utils

然后去python测试pytorch版本：

(uutils_env) miranda9~/type-parametric-synthesis $ python
Python 3.9.7 (default, Sep 16 2021, 13:09:58) 
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torchtext

>>> 
>>> from torchtext.vocab import Vocab
>>>

它知道根据我的 setup.py 文件安装正确的版本。但是你可以安装正确的版本如下：

pip install torchtext==0.10.1

将来上述版本可能会发生变化，您可能需要在 torchtext 的 github.

中提出问题

注：

如果您使用的是 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

但没试过。

Acks：特别感谢@ivan 的帮助！

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

How does one install torchtext with cuda >=11.0 (and pytorch 1.9)?

pytorch