每個入坑深度學習的小白都可能面臨TensorFlow的安裝問題,在安裝時首先面臨的就是版本選擇的問題:CPU版還是GPU版?版本號選哪個?
1 GPU版?CPU版?
判斷條件 | 結果 |
---|---|
顯卡是否NVIDIA系列? | 是=GPU;否=CPU |
若是NVIDIA系列,計算能力如何? | 大於等於3.5=GPU;小於3.5=CPU |
解釋:
- 首先,查看自己電腦顯卡的型號(詳見:如何查看及安裝)。如果顯卡是NVIDIA系列的,繼續下面步驟;如果顯卡不是NVIDIA系列的,直接裝CPU版。
- 然後,如果是NVIDIA系列的,則查詢該顯卡的計算能力。詳見:查詢計算能力。點開自己顯卡對應的系列,查看自己顯卡的GPU計算能力(即,Compute Capability)。如下圖,顯卡NVIDIA Quadro RTX 8000對應的計算能力爲7.5。
- 最後,到官網查詢發佈的GPU支持,查看硬件要求。本文查詢時的標準是3.5(如下圖)。如果計算能力≥3.5,可以裝GPU版;相反<3.5的只能選擇CPU版了。
ps: CPU版和GPU版兩者沒啥大差別,只是計算快慢的問題(呵呵~)。