深度学习环境配置
主要记录一下在 Ubuntu16.04 中配置 tensorflow-gpu+opencv的过程。
安装Cuda
Attention 安装驱动前需要先确定所需要的cuda版本,根据cuda版本选择驱动版本,根据驱动版本选择系统的内核版本。
查看Tensorflow 对应的cuda版本: https://tensorflow.google.cn/install/source#linux
查看cuda对应的驱动版本:https://developer.nvidia.com/cuda-toolkit-archive 查看指定cuda版本的online documentation 的 installation guide 一栏,确定要求的系统内核版本。
一般对于Ubuntu16.04,cuda-10 及其以下版本都只支持4.4.0版本的内核,而目前Ubuntu官网提供的最新的16.04的iso文件内核版本为4.15.0 (查看系统内核版本命令 uname -r
)。可以使用降低内核来达到内核版本要求(但是我没成功)。另外一种方式就是使用Ubuntu16.04.1 重新安装系统,该系统自带内核版本为 4.4.0 。
另外,推荐安装完系统后安装Anaconda用来管理Python开发环境。
安装步骤
- 禁用nouveau