unilm ERROR: Could not find a version that satisfies the requirement torch==1.7.1+cu110

cx6n0qe3 于 2个月前发布在其他

关注(0)|答案(2)|浏览(26)

描述bug

我使用的模型(UniLM,MiniLM,LayoutLM等):
当使用以下情况时出现问题：

官方示例脚本：(详细说明如下)
我自己的修改过的脚本：(详细说明如下)

conda create -n trocr python=3.7
conda activate trocr
git clone https://github.com/microsoft/unilm.git
cd unilm
cd trocr
pip install pybind11
pip install -r requirements.txt
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" 'git+ https://github.com/NVIDIA/apex.git '
一个清晰简洁的描述bug是什么。

重现bug

重现行为所需的步骤：
1.
2.
3.

预期行为

一个清晰简洁的描述你期望会发生什么。

平台：OSX 16
Python版本：3.8.5
PyTorch版本(GPU?):

unilm

来源：https://github.com/microsoft/unilm/issues/469

2条答案

按热度按时间

y3bcpkx11#

请运行以下命令，并告诉我是否成功或失败：

pip install pybind11
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 -c pytorch
pip install timm==0.4.5
pip install natsort
pip install tensorboard
pip install nltk
pip install h5py
pip install numpy
pip install fastwer
pip install git+https://github.com/liminghao1630/fairseq.git

赞(0）回复(0）举报 2个月前

cgvd09ve2#

这对layoutLM模型有效。

赞(0）回复(0）举报 2个月前