TENSORFLOW GPU在Windows下的安裝 === 如果你的電腦配有一片(或多片)Nvidia的GPU顯示卡,那你有很大的機會可以利用到GPU來加速Tensorflow模型的訓練及預測。Tensorflow GPU的安裝除了硬體之外,你也必須安裝上對應的驅動程式和函數庫。如果你的Windows已經將Nvidia的驅動程式安裝完畢,驅動程式這部分就算安裝完成,接下來要做的就是安裝CUDA的函數庫及CUDNN來幫助深度學習神經網路的加速。相關工具的版本必須同時符合TensorFlow的版本、NVidia顯卡的限制,否則執行時會有錯誤發生,當TensorFlow升級時,版本也要隨之改變。 你可以使用 ``` conda install tensorflow-gpu ``` 或是 ``` pip install tensorflow-gpu ``` 當你的環境安裝完Tensorflow之後,就可以正確使用GPU來加速了。在程式中輸入: ```python= import tensorflow ``` 如果沒有出現錯誤訊息,表示Tensorflow安裝完畢了。