问题
明明安装了tf-gpu但是训练时还是使用cpu
解决办法
- 先查看tensorflow可用设备
>>python
>from tensorflow.python.client import device_lib
>print(device_lib.list_local_devices())
发现真的只有cpu了。。。。
- pip list发现虚拟环境既装了CPU版本又安装了GPU版本,而且 cpu版本还不知被谁装在base环境里了。。。
- 卸载所有tf,重新安装gpu版本(记得对应cuda)
然后就可以了。。。。