在使用transformers和pytorch时出现的版本冲突的问题

作者：煮酒与君饮 | 2024-08-18 18:06:42

踩

在使用transformers和torch库的时候，出现了以下问题：

1、OSError: [WinError 126] 找不到指定的模块。 Error loading "D:\Program Files\anaconda3\envs\testenv\Lib\site-packages\torch\lib\fbgemm.dll" or one of its dependencies.

2、error: bad escape \P at position 28；

3、ImportError: cannot import name 'dummy_torchvision_objects' from 'transformers.utils' (D:\Program Files\anaconda3\envs\testenv\Lib\site-packages\transformers\utils\__init__.py)；

4、RuntimeError: Failed to import transformers.models.bert.modeling_bert because of the following error (look up to see its traceback): No module named 'safetensors.torch'；

这些都是transformers和pytorch版本不匹配造成的问题，并没有找到transformers对pytorch等的版本要求，但是因为两种框架的更新是不同步的，如果直接使用它们的最新版本，还是很容易出现冲突的问题。

下面我提供一个windows上使用后不会冲突的版本组合，供大家参考：

pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 transformers==4.39.3

如果还是使用了numpy，建议使用numpy<2.0的版本

pip install numpy==1.26.4

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/煮酒与君饮/article/detail/998521