如何选择安装的TensorFlow版本?

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

1 GPU版?CPU版?

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

解释:

  1. 首先,查看自己电脑显卡的型号(详见:如何查看及安装)。如果显卡是NVIDIA系列的,继续下面步骤;如果显卡不是NVIDIA系列的,直接装CPU版。
  2. 然后,如果是NVIDIA系列的,则查询该显卡的计算能力。详见:查询计算能力。点开自己显卡对应的系列,查看自己显卡的GPU计算能力(即,Compute Capability)。如下图,显卡NVIDIA Quadro RTX 8000对应的计算能力为7.5。
    在这里插入图片描述
  3. 最后,到官网查询发布的GPU支持,查看硬件要求。本文查询时的标准是3.5(如下图)。如果计算能力≥3.5,可以装GPU版;相反<3.5的只能选择CPU版了。
    在这里插入图片描述
    ps: CPU版和GPU版两者没啥大差别,只是计算快慢的问题(呵呵~)。

2 版本号选哪个?

2.1 CPU版

Windows:
在这里插入图片描述Linux
在这里插入图片描述macOS
在这里插入图片描述

2.2 GPU版

Windows:
在这里插入图片描述Linux
在这里插入图片描述macOS
在这里插入图片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章