如何使用 cuda >=11.0(和 pytorch 1.9)安装火炬文本？

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

使用空env应该很容易复制，如下所示:

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

crossposted:

• https://discuss.pytorch.org/t/how-does-one-install-a-torchtext-version-compatible-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.2andCUDA 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

对于我来说，它适用于torchtext0.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/中最新的命令，但如果这不起作用，请转到torchtext网站查看他们支持的python和pytorch的版本并安装。)希望将来torchtext能与pytorch主分支保持一致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的帮助!