TensorFlow 1.5导入错误:找不到tensorflow.python.client模块求助
解决TensorFlow 1.5.0导入
tensorflow.python.client的错误 看起来你遇到的ImportError大概率是环境依赖或者包安装的问题,结合你用的TensorFlow 1.5.0版本,我给你几个排查和解决的方向:
1. 检查当前环境的TensorFlow包情况
首先确认你的Python环境里是不是真的正确安装了tensorflow-gpu,有没有和普通tensorflow包冲突。运行下面的命令:
pip list | grep tensorflow
如果输出里同时出现tensorflow和tensorflow-gpu,建议卸载其中一个,优先保留tensorflow-gpu:
pip uninstall tensorflow
2. 确认虚拟环境(如果使用)是否激活
如果你是用虚拟环境(比如venv、conda)安装的tensorflow-gpu,但运行脚本时没激活对应的环境,Python会用系统默认的环境,自然找不到这个包。先激活你的虚拟环境再运行脚本试试。
3. 调整导入语句的写法
有时候直接导入子模块会出问题,你可以先导入TensorFlow主模块,再导入device_lib:
import tensorflow as tf from tensorflow.python.client import device_lib def get_available_gpus(): local_device_protos = device_lib.list_local_devices() return [x.name for x in local_device_protos if x.device_type == 'GPU'] print(get_available_gpus())
4. 重新安装TensorFlow-GPU 1.5.0
可能安装过程中包文件损坏了,先卸载再重新安装:
pip uninstall tensorflow-gpu pip install tensorflow-gpu==1.5.0
如果是Python3环境,记得把pip换成pip3。
5. 检查CUDA和cuDNN版本是否匹配
TensorFlow 1.5.0对CUDA和cuDNN版本有严格要求:需要CUDA 9.0和cuDNN 7.0,如果你的版本不符合,不仅会导致导入错误,后续也无法正常使用GPU。可以去NVIDIA官网下载对应版本的CUDA和cuDNN,配置好环境变量后再测试。
内容的提问来源于stack exchange,提问作者Magick
相关产品推荐
相关产品推荐

