TensorFlow GPU Windows 配置

"TensorFlow-GPU for Windows10"

Posted by Canary on May 26, 2018

前言

好机器不能浪费,遂在本人笔记本 DELL 5530 上安装 TensorFlow-GPU 。

环境

  • Windows10家庭版
  • GPU Nvidia M1200
  • Python 3.5.3
  • CUDA 8.0
  • cuDNN 5.0.5

资源地址

  1. cuda_8.0.61_win10.exe下载地址。cuda_8.0.61_win10

    安装完成后,在系统环境变量会有如下环境变量:

     CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
     CUDA_PATH_V8_0=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
     NVCUDASAMPLES8_0_ROOT=C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0
     NVCUDASAMPLES_ROOT=C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0
     NVTOOLSEXT_PATH=C:\Program Files\NVIDIA Corporation\NvToolsExt\
    
  2. cuDNN 下载需要注册账号。就不提供地址了,认准版本号就可以。下载完之后,将压缩包解压到CUDA安装目录中,比如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0;或者定义CUADD_ROOT缓存变量指向解压缩cuDNN文件的位置。

  3. Python 3.5.3 我使用了Anaconda3,通过创建虚拟环境安装,因为本人电脑已经安装有Caffe,如果两个框架安装在同一环境中,生怕出事,于是新建环境。

     conda create -n tensorflow python=3.5.3
    

    然后将新创建的虚拟环境添加到Path环境变量:

     C:\ProgramData\Anaconda3\envs\tensorflow
     C:\ProgramData\Anaconda3\envs\tensorflow\Scripts
    

    两个常用指令:

     打开环境:activate tensorflow
     关闭环境:deactivate tensorflow
    

使用 Anaconda navigator 安装 TensorFlow-GPU

使用搜索框检索 “tensorflow-gpu” 然后点击安装即可。

验证安装

打开anaconda prompt,激活环境并进入python:

测试代码如下:

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

运行结果:

至于中间那一串警告,只是框架的一个建议而已:

The TensorFlow library wasn't compiled to use SSE instructions, but these are available on your machine and could speed up CPU computations.

忽略之即可!

知识共享许可协议本作品采用知识共享署名-相同方式共享 4.0 国际许可协议进行许可。欢迎转载,并请注明来自:黄钢的博客 ,同时保持文章内容的完整和以上声明信息!