tensorflow-gpu找不到Gpu device的解决方法记录

问题

明明安装了tf-gpu但是训练时还是使用cpu

解决办法

1. 先查看tensorflow可用设备

>>python
>from tensorflow.python.client import device_lib
>print(device_lib.list_local_devices())

发现真的只有cpu了。。。。

2. pip list发现虚拟环境既装了CPU版本又安装了GPU版本，而且 cpu版本还不知被谁装在base环境里了。。。
3. 卸载所有tf，重新安装gpu版本（记得对应cuda）

然后就可以了。。。。