查看本机GPU是否支持安装tensorflow-gpu
安装gpu版本的tensorflow需要安装cuda和cudnn
所以需要检查cuda是否支持本机gpu,需要计算能力大于3.1
1、查看本机显卡配置,对比官网的显卡型号是否支持
显卡型号支持:https://developer.nvidia.com/cuda-gpus
显卡配置为:NVIDIA DeForce GTX 1050
可知计算能力6.1
支持安装
2、参考链接
https://blog.csdn.net/Eppley/article/details/79297503