tensorflow 之 ubuntu 18.04 安装
1. tensorflow 要求环境
- 平台:Ubuntu 18.04
- 显卡:nvidia gtx 1060
- 平台:x86_64
- cuda版本:10.2
2. 安装显卡驱动
- Ctrl + Alt + t 打开终端,输入更新软件列表: sudo apt-get update
- 打开【软件和更新】-【附加驱动】- 选择驱动更新,然后重启电脑
- 在终端中输入:nvidia-settings,打开显卡设置界面即表示安装成功
3. 安装cuda
cuda是nvidia的并行运算的软件包,使用GPU能够极大提高神经网络的的运算速度。
3.1 cuda的特点
- 提供并行运算的标准语言程序,如c
- 支持同时进行串行计算与并行计算:串行计算在CPU上计算,并行计算在GPU上进行
3.2 安装前准备
由于 cuda 安装需要相应的安装与运行环境,因此,再进行cuda安装前进行环境的检查。环境检查主要包含如下几个部分:
- 当前 GPU 是否支持所要安装的cuda版本
- 当前系统是否支持
- 当前 kernel header 与环境是否支持
- gcc 的版本是否支持
- glibc 的版本是否支持
3.2.1 当前 GPU 是否支持所要安装的cuda版本
- Ctrl + Alt + t 打开终端,查看显卡版本:lspci | grep -i nvidia
- 如果什么都没有返回,个更新 pci 数据库,使用命令:updta-pciids
- 如果有返回内容,则查看 显卡版本是否在安装的 cuda 版本列表中
3.2.2 当前系统是否支持
- Ctrl + Alt + t 打开终端,查看系统版本:uname -a 或者 cat /proc/version
- 然后根据系统版本,选择cuda的版本
3.2.3 当前 kernel header 与环境是否支持
- Ctrl + Alt + t 打开终端,查看 kernel header 的版本:uname -m &&