Install TensorFlow with GPU on Windows

Windows10 64 bits + Visual Studio 2015 + CUDA8.0 + TensorFlow with GPU#

终于把TensorFlow-gpu 安装好了,记录下自己的安装过程,希望他人能少走弯路。
参考的安装教程:

  • TensorFlow官网
  • 上的一篇教程
  • github的一篇教程

Install CUDA8.0##

我直接安装CUDA8.0 version的时候在安装DisplayDriver的时候遇到NVIDIA installer failed的问题,有可能是和我之前安装过DisplayDriver有关。Anyway,我尝试重新安装DisplayDriver,然后安装CUDA的时候选择Customer Version,不选择DisplayDriver有关的components。具体如下:

  • Download and Install DisplayDriver
    安装相应的版本(我的是GPU是GTX1070,下载链接),选择Customer Version,并且勾选perform a clean installation
  • Download and Install CUDA8.0
    Toolkit version 8.0 or above: https://developer.nvidia.com/cuda-downloads
Install TensorFlow with GPU on Windows_第1张图片
CUDA8.0

选择Customer Version,不要选择DisplayDriver Component

  • Download and install cuDNN
    cuDNN version 5.1 library for Windows 10: https://developer.nvidia.com/cudnn
    You would need to signup at Nvidia in order to download these files. Now extract the cuDNN files into your Toolkit directory.
Install TensorFlow with GPU on Windows_第2张图片
cuDNN
  • Environmental variables
    参考 https://nitishmutha.github.io/tensorflow/2017/01/22/TensorFlow-with-gpu-for-windows.html
  • 测试CUDA安装结果,编译CUDA示例程序
    参考 http://www.jianshu.com/p/c245d46d43f0

Install TensorFlow with GPU##

  • 安装Anaconda和TensorFlow with GPU
    参考 http://www.jianshu.com/p/c245d46d43f0 ,采用离线方式安装

写的很凌乱,有时间再来整理。。。

你可能感兴趣的:(Install TensorFlow with GPU on Windows)