版本眾多混亂,那該怎麼選擇適合、可運行的TenorFlow 版本呢?
會需要使用到CPU- Z,請先下載。
CPU- Z:https://www.cpuid.com/softwares/cpu-z.html
開啟CPU-Z,點選Graphics查看顯示卡是否為nVIDIA。
如果為否,則安裝CPU版。反之,則可以安裝GPU版加速。
接著點開CPU的部份,查看是否有AVX的指令集。
如果為否,安裝1.6.0版會出bug,需要安裝1.5.0(含以前)的版本。如果有支援AVX則可安心使用1.6.0版。
綜合以上所述:
是否為Nvidia之顯卡
|
|||
是
|
否
|
||
CPU是否支援AVX指令集
|
是
|
GPU
1.6.0
|
CPU
1.6.0
|
否
|
GPU
1.5.0
(含以下)
|
CPU
1.6.0
|
各版本在Python 3.6版64bit之安裝指令:
GPU 1.6.0 :
pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.6.0-cp36-cp36m-win_amd64.whl
GPU 1.5.0:pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.5.0-cp36-cp36m-win_amd64.whl
CPU 1.6.0:pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.6.0-cp36-cp36m-win_amd64.whl
CPU 1.5.0:pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.5.0-cp36-cp36m-win_amd64.whl
相關安裝請參考:於Win10下安裝TensorFlow並用GPU做加速