win10系统安装tensorflow-gpu版本

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

本人使用的win10  64位   专业版    python3.5  

  1. pip install tensorflow-gpu

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

      下载地址: 点击打开链接

       安装CUDA8,直接安装好就可以,环境变量会自动添加。

下面很重要:

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

win10系统安装tensorflow-gpu版本_第1张图片win10系统安装tensorflow-gpu版本_第2张图片
win10系统安装tensorflow-gpu版本_第3张图片


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

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


win10系统安装tensorflow-gpu版本_第4张图片




你可能感兴趣的:(win10系统安装tensorflow-gpu版本)