首先声明:以下均为 Linux(Ubuntu 14.04)环境下,且采用 miniconda(anaconda也可以)进行安装
之前配置过一次,虽然最终配好了,但是稀里糊涂的,尤其是CUDA、cudnn、tensorflow 三者版本的问题,之前的配置过程可以看这里。
前几天到了新公司,又要配置了,折腾了半天终于算是搞明白应该怎么配置了。
顺便提一下,感觉 tensorflow 官网的这个版本图不太准,而且没有说明正确的安装顺序。
下面进入正题
0、要安装 tensorflow-gpu ,需要处理以下几个依赖:按顺序依次为 python、CUDA driver、CUDA toolkit、cudnn;
1、python 版本
现在下载 anaconda/miniconda 默认是3.7版本的,可能会有很多问题,所以最好是新开一个 python3.6 的虚拟环境
conda create -n tf36 python=3.6
其中 tf36 是环境名