训练中的错误OpenNMT -Caffe2_detectron_ops.dll找不到

我有火炬 1.6 和 python 3.8。在训练 OpenNMT 时,它会抛出以下错误 -

OSError: [WinError 126] 找不到指定的模块。加载“C:UsersGirishAppDataLocalProgramsPythonPython38libsitepackagestorchlibcaffe2_detectron_ops.dll”或其依赖项之一时出错。

我检查了文件夹,文件在那里。我试过卸载torch并重新安装它,但没有帮助。

任何帮助将不胜感激。谢谢

stack overflow Error in training opennmt - caffe2_detectron_ops.dll not found
原文答案

答案:

作者头像

对我来说同样的错误,但我认为删除这些项目只会禁用 GPU 使用,这不是我想要的。我试过了,错误消失了,但是使用os GPU也...

import torch
torch.cuda.is_available()

返回为假。

重新安装手电筒对我有用:

# pip3 install torch==1.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html --user
# pip3 install torchaudio===0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html --user
# pip3 install torchvision==0.11.2+cu113 torchaudio===0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html --user

相关问题