如何選擇安裝的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
在這裏插入圖片描述

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