Ubuntu18.04,CUDA10.0,TensorFlow-GPU安装

August 30, 2019

开始以下步骤之前,强烈建议先参考 Ubuntu18.04安装NVIDIA显卡驱动 正确安装显卡驱动。如果显卡驱动已经安装则下面安装 CUDA10 流程可以省略,因为 CUDA10 会自动装好。

Python环境

Ubuntu18.04 同时包含 Python2Python3,但 Python3 相关环境例如 pip 不一定有,所以需要安装

1
sudo apt-get install python3-pip python3-dev python-virtualenv

CUDA10 & cuDNN

cuda-10.0-download-archive 获取 CUDA10,如果已经安装可以跳过。

cuda_10

用以下命令打开安装包

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)

cuDNN_7_6_1

安装

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 即可