tensorflow突然检测不到GPU,显示False的问题
问题描述:
之前装好后输入以下命令,可以正常显示以下内容;但是之后重新安装了TensorFlow-GPU后测试,检测不到GPU,显示false的问题。
# 检测 tensorflow 能使用的设备情况
from tensorflow.python.client import device_lib
#os.environ["CUDA_VISIBLE_DEVICES"] = "1" # 这个可以指定使用哪个设备
print(device_lib.list_local_devices())

问题解决:
1.先激活所在环境activate tf-gpu-py38
2.执行命令pip uninstall protobuf
3.2成功后继续执行命令pip uninstall tensorflow
4.再次进行TensorFlow-GPU的安装,执行命令pip install tensorflow-gpu==2.3.0(最好不要使用conda进行安装)
5.安装成功后进行测试,全部正常显示!OK,搞定!!!
版权声明:本文为weixin_47274607原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。