No module named 'tensorflow.io'询问

cmd端使用pip install tensorflow.io后,下载的同时更新tensorflow至2.5,虽然能正常使用但计算时选择的是2.5版本cpu的tensorflow。

此时导入如下代码

import tensorflow_datasets as tfds

出现“Could not load dynamic library 'cudart64_101.dll”

所以我将tensorflow2.5降低到2.1后,gpu版本又能正常使用了,不过又重新显示如下错误,请问是什么原因呢?百思不得其解,希望有大神来解答

 

ModuleNotFoundError: No module named 'tensorflow.io'

 

版本问题,2.x好像没有这个了,pip安装一下看看?

pip install tensorflow.io

原来是我下载的tensorflow-datasets版本太高了,降级之后就能使用了