开始以下步骤之前，强烈建议先参考 Ubuntu18.04安装NVIDIA显卡驱动 正确安装显卡驱动。如果显卡驱动已经安装则下面安装 CUDA10 流程可以省略，因为 CUDA10 会自动装好。
Python环境
Ubuntu18.04 同时包含 Python2 和 Python3，但 Python3 相关环境例如 pip 不一定有，所以需要安装
1
sudo apt-get install python3-pip python3-dev python-virtualenv
CUDA10 & cuDNN
从 cuda-10.0-download-archive 获取 CUDA10，如果已经安装可以跳过。
用以下命令打开安装包
1
2
3
4
sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
注册账号登录到 developer.nvidia.com，下载 cuDNN Runtime Library for Ubuntu18.04 (Deb)
安装
1
sudo dpkg -i libcudnn7_7.6.1.34-1+cuda10.0_amd64.deb
为 ~/tensorflow 设置 virtualenv
1
virtualenv --system-site-packages -p python3 ~/tensorflow
调用脚本激活
1
2
source ~/tensorflow/bin/activate
(tensorflow)
在相同终端内更新 pip
1
easy_install -U pip
下载 tensorflow-gpu，国内指定临时下载源: https://pypi.tuna.tsin.edu.cn/simple 加快下载速度
1
pip3 install --upgrade tensorflow-gpu -i https://pypi.tuna.tsin.edu.cn/simple
最后运行 TensorFlow 即可