环境搭建04——如何选择安装的TensorFlow版本？

最新推荐文章于 2025-02-23 22:31:25 发布

命名无能

最新推荐文章于 2025-02-23 22:31:25 发布

阅读量6.6w

点赞数 63

分类专栏： # 系统与环境搭建文章标签： TensorFlow安装 TensorFlow版本选择

本文链接：https://blog.csdn.net/qiancaobaicheng/article/details/95226499

版权

每个入坑深度学习的小白都可能面临TensorFlow的安装问题，在安装时首先面临的就是版本选择的问题：CPU版还是GPU版？版本号选哪个？

判断条件	结果
显卡是否NVIDIA系列？	是=GPU；否=CPU
若是NVIDIA系列，计算能力如何？	大于等于3.5=GPU；小于3.5=CPU

解释：

首先，查看自己电脑显卡的型号（详见：如何查看及安装）。如果显卡是NVIDIA系列的，继续下面步骤；如果显卡不是NVIDIA系列的，直接装CPU版。
然后，如果是NVIDIA系列的，则查询该显卡的计算能力。详见：查询计算能力。点开自己显卡对应的系列，查看自己显卡的GPU计算能力（即，Compute Capability）。如下图，显卡NVIDIA Quadro RTX 8000对应的计算能力为7.5。