查看本机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

本文介绍如何检查本机GPU是否支持安装TensorFlow-GPU版本,包括验证GPU计算能力和查阅支持的显卡型号。以NVIDIA GeForce GTX 1050为例,其计算能力为6.1,满足安装条件。

2862

被折叠的 条评论
为什么被折叠?



