Ubuntu 16.04+Gtx1050Ti+cuda 8.0+cudnn 5.1 tensorflow 安装
tensorflow
一个面向所有人的开源机器学习框架
项目地址:https://gitcode.com/gh_mirrors/te/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卡的驱动,就不要按照官网的来进行安装了,安装的时候,后面不写参数,然后到第二步骤的时候,选择不安装驱动,如果选择安装的话,会出错。
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 个月前
更多推荐
已为社区贡献3条内容
所有评论(0)