---
# System prepended metadata

title: TENSORFLOW GPU在Windows下的安裝

---

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安裝完畢了。

