您的位置:首页 > 其它

[置顶] win10系统安装tensorflow-gpu版本

2018-01-07 19:43 766 查看
本人在CSDN的第一篇博客!!!   之前因为要学习tensorflow因此一直想在自己电脑上安装tensorflow,一开始图简单安装的CPU版的,但是发现只要跑稍微大点儿程序,动不动就要跑5,6个小时,因此一直想装GPU版的tensorflow但是一直配置不好环境,经过无数次的尝试,终于在自己电脑上配置成功。现将配置方法分享给大家!
本人使用的win10  64位   专业版    python3.5  

pip install tensorflow-gpu

安装CUDA8+cudnn   这里cudnn要安装两个版本。

      下载地址:点击打开链接
       安装CUDA8,直接安装好就可以,环境变量会自动添加。

下面很重要:

 将解压出来的cuda文件夹下的bin、include、lib中的文件复制到C盘中安装CUDA8相应的文件夹下;











这里需要复制两个文件夹中bin   include  lib中的文件(除了bin中的文件不一样外,其他都一样,一个文件对应的是cudnn5.1,另一个是cudnn6)。

     这样就完成了tensorflow-gpu版本的环境配置。

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: