執行完之前介紹的 為 Ubuntu 18.04 LTS 安裝 Nvidia GPU driver 與 為 Ubuntu 18.04 LTS 安裝 Python 3.6 後，就可以為我們的作業系統安裝能透過 GPU 運算的 TensorFlow 啦。

當前的 TensorFlow 版本為 1.12，只支援 CUDA 9.0，當前 CUDA 9.0 的 cuDNN 版本為 7.4.2。

安裝 CUDA

原則上，Ubuntu 18.04.1 LTS 已經內建 Python 3 了，保險起見先檢查一下：

1
2
3

sudo apt update
sudo apt upgrade
python3 -v

此時的 Python 的版本應有 3.6 以上，我的版本為 3.6.7。

安裝 pip 3

列出支援的顯示卡驅動：

1

ubuntu-drivers devices

我這邊使用 Nvidia GTX 1060 6G，當前的驅動版本為 415，若是列出的版本太舊(390)，請執行下面這行把 Nvidia Driver 加入 apt 的來源中：

1
2
3

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
ubuntu-drivers devices

看看是否有更新的驅動程式版本可以使用。