Ubuntu18安裝Tensorflow (gpu)

首先先update一下

sudo apt-get update

sudo apt-get upgrade

新增顯示卡repository​

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt-get update

安裝dependence

sudo apt-get install dkms synaptic build-essential

打開“software & update”->”Additional drivers”, 選擇需要安裝的driver, 安裝後重啟

重啟後可運行“nvidia-smi” 檢視顯示卡狀態

另外，可運行以下命令檢查溫度, (1 = 每秒刷新)

watch -n 1 nvidia-smi

安裝 cuda

到官方網站選擇以network方式安裝

sudo dpkg -i cuda-repo-ubuntu1810_10.1.105-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1810/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

安裝CUDNN

下載CUDNN(需要註冊)

解壓並複製對文件到CUDNN對應資料夾內

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

安裝Tensorflow-gpu

pip3 install tensorflow-gpu

安裝完成後可以打開python import一下測試

python3

import tensorflow

如沒有報錯即安裝成功。