Ubuntu18.04发行已经有一段时间了,正好最近Tensorflow也发布了1.8版本,于是决定两个一起装上,以下是安装总结,大致可以分为5个步骤
确认当前软件和硬件环境、版本
更新显卡驱动,软件版本准备
CUDA 9.0 ToolKit安装
cuDNN7.1.3 for CUDA9.0安装
TensorFlow GPU 安装
Test it!
1.确认硬件软件环境、版本
系统版本,Ubuntu18.04 自然没什么好说的, 使用指令sudo lsb_release -a ,得到以下输出结果
No LSB modules are available.
Distributor ID:Ubuntu
Description:Ubuntu
Codename:bionic
GCC和G++ 版本,18.04的ubuntu默认的是7.0,同时也有附带安装6.0,不过我们这次安装需要更低版本的GCC以及G++
gcc --version 查看GCC版本号 gcc (Ubuntu 7.0 -4ubuntu8) 7.0
g++ --version 查看G++版本号 g++ (Ubuntu 7.0-4ubuntu8) 7.0
1
2
3
我选择采用的是4.8版本gcc和g++,后面给出降级方法。
英伟达显卡驱动版本, 使用nvidia-smi 可以得到相关信息,我使用的是GTX970显卡,驱动已经更新到390.48版本。
Python 版本, python2 –version 和 python3 –version, 应该对应2.7+ 和 3.6+版本了都