python 既没有找到PyTorch也没有找到TensorFlow >= 2.0,模型将不可用,只能使用标记器,配置和文件/数据实用程序

yjghlzjz  于 2023-05-21  发布在  Python
关注(0)|答案(3)|浏览(705)

我正在尝试使用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
yizd12fk

yizd12fk1#

你需要一个PyTorch或Tensorflow。
您可以检查是否安装了TensorFlow,也可以重新安装

  1. pip uninstall tensorflow
  2. pip install tensorflow==2.2.0(你只能安装tensorflow,它的工作原理和tensorflow-gpu一样)
  3. pip uninstall transformer
  4. pip install transformer==3.3.1
    如果这不能解决问题,请尝试将您的python升级到3.7.8
    希望它能解决你的问题。
h7wcgrx3

h7wcgrx32#

我调查了10个小时才发现问题
我使用conda install tensorflow-gpu安装了tensorflow
在拆除tensorflow-gpu并使用pip安装后,使用pip安装变压器
它工作正常

jtw3ybtb

jtw3ybtb3#

你好,只是想添加我使用的是M1 Mac,也遇到了同样的问题。对于M1 mac(或更高版本)实际上是不同的包。下面的requirments.txt对我来说很有用(只需要python 3.10 + venv):
tensorflow-macos==2.12.0
tensorflow-metal==0.8.0
变压器==4.29.2

相关问题