安装tensorflow的GPU版本需要安装CUDA和cuDNN,所以要检查本机的GPU是否支持CUDA,计算能力大于3.1才可以。
查看本机显卡配置
右键我的电脑——管理——设备管理器
可以看到显卡配置为:NVIDIA GeForce GTX 1050
显卡型号支持官网:https://developer.nvidia.com/cuda-gpus 传送门
对比官网型号,计算能力6.1,支持安装。
安装CUDA和cuDNN可以参考这篇→https://blog.csdn.net/qq_43627106/article/details/124175225