赞
踩
以前安装的是tensorflow-cpu,但是随着运算量的提高产生了鼓捣tensorflow-gpu的想法。官网的教程看起来很简单,但是坑很多。本文主要是帮助大家绕过那些坑,成功安装可以运行的tensorflow-gpu。
下载cuDNN v6.0,注意下载这个安装包需要注册并且填一堆问卷,下下来以后解压,然后直接拷到cuda路径对应的文件夹下面就行。具体操作是讲将下图对应三个文件夹里面的文件分别拷贝到cuda安装目录对应文件夹
cuda安装目录如下图所示
如果现实如图所示就证明tensorflow-gpu安装成功了
如果不明白的话,大家可以参考点击
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。