Ubuntu 16.04+Gtx1050Ti+cuda 8.0+cudnn 5.1 tensorflow 安装


现在tensorflow 终于支持高版本(cuda 8.0)一键安装了,今天尝试了一下,很爽。不用折腾那么多东西。

建议安装 anaconda ,一个特别好的Python包管理器,清华大学的源有其镜像,安装速度,更新也挺快的。 其独有的虚拟环境可以将一个个环境隔离开来,互不影响,这个特别好。

我们首先安装anaconda ,这个官网有教程,整个安装过过程傻瓜化。
首先创建虚拟环境:
conda create -n tensoflow python=3.7
conda create -n tensoflow python=3.4
conda create -n tensoflow python=3.5

根据习惯选一个Python版本。

tf的gpu版本现在支持直接用pip直接安装,前提是安装好cuda8.0和cudnnv5.1(必须是8.0和5.1不然会报错),然后直接在命令行下pip install tensorflow-gpu

这里说一下cuda 的安装,官网给出的安装教程前提是没有安装官方N卡驱动的。按照官网方法一步步来可以安装成功,并且很快,不需要选择就可以安装好,然后根据CUDA 的sample进行测试。一般都会安装成功

如果之前单独安装过N卡的驱动,就不要按照官网的来进行安装了,安装的时候,后面不写参数,然后到第二步骤的时候,选择不安装驱动,如果选择安装的话,会出错。