尚未找到 PyTorch 和 TensorFlow >= 2.0。模型将不可用，只能使用分词器、配置和文件/数据实用程序

Question

我正在尝试使用 pip 安装变压器

pip install transformers

进口变压器后

这个错误显示

Neither PyTorch nor TensorFlow >= 2.0 have been found.Models won't be available and only tokenizers, configuration, and file/data utilities can be used.

虽然我安装 TensorFlow-GPU= 2.3.1 并使用 conda

系统信息

Windows 10 
python 3.6
cuda 10.1
tensorflow-gpu= 2.3.1

Answer 1

您需要 PyTorch 或 Tensorflow 之一。

您可以检查是否安装了tensorflow或者可以重新安装

1. pip uninstall tensorflow
2. pip install tensorflow==2.2.0（你可以只安装tensorflow，它的工作方式与tensorflow-gpu相同）
3. pip uninstall transformers
4. pip install transformers==3.3.1

如果这不能解决问题，请尝试将 python 升级到 3.7.8

Answer 2

经过10个小时的调查，我发现了问题

我通过使用安装了张量流conda install tensorflow-gpu

删除tensorflow-gpu并使用pip安装后，使用pip进行变压器

效果很好