现在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卡的驱动,就不要按照官网的来进行安装了,安装的时候,后面不写参数,然后到第二步骤的时候,选择不安装驱动,如果选择安装的话,会出错。


GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2 PiperOrigin-RevId: 663726708 2 个月前
91dac11a This test overrides disabled_backends, dropping the default value in the process. PiperOrigin-RevId: 663711155 2 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐