要搭建TensorFlow的GPU版本，首先需要的必備條件就是一塊能夠支持CUDA的NVIDIA顯卡，因爲在搭建TensorFlow的GPU版本時，首先需要做的一件事就是安裝其基礎支持平臺CUDA和其機器學習庫cuDNN，然後在此基礎上搭建TensorFlow GPU版本。
其次還要了解一下不同的TensorFlow版本所需要對應安裝的CUDA和cuDNN版本是多少，因爲在TensorFlow的GPU版本安裝過程中，如果對應的CUDA版本和cuDNN版本不正確的話，是無法正常使用GPU來進行模型訓練的。下表整理出了TensorFlow從1.2到最新版本的CUDA和cuDNN所對應的版本集合。
ref:參考了文章－AI的那些事兒