1.查看电脑显卡

路径:计算机--属性--设备管理器-显示适配器

2.显卡(GPU)是否支持CUDN

https://developer.nvidia.com/cuda-gpus

3.安装

1)CUDA Toolkit版本下载:https://developer.nvidia.com/cuda-toolkit-archive

    我下载的是:cuda_9.0.176_windows

    下载完后进行安装,一路确认到底。

2)cuDNN版本下载:https://developer.nvidia.com/rdp/cudnn-download(先注册再下载)

     我下载的是:cudnn-9.0-windows7-x64-v7.1(对应cuda9.0版本

     下载完后,将文件解压;

     在C盘根目录下新建个tools文件夹,将解压后的“cuda”文件夹放入其中

     设置环境变量:进入“Path”,添加"C:\tools\cuda\bin"与”C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin”

3)Anaconda版本下载:https://repo.continuum.io/archive/

     1.我下载的是Anaconda3-5.1.0-windows-x86_64(这里安装包里默认的是Python3.6, Anaconda2-5.1.0-windows-x86_64默认安装的是Python2.7)

     在安装过程中:选择“just me”和“add anaconda to my path...”

     2.创建tensorflow的conda环境

C:>conda create -n tensorflow pip python=3.5

     备注:安装过程中选择yes

     3.激活conda环境

C:>activae tensorflow  

4)TensorFlow-gpu安装

      版本下载:https://pypi.python.org/pypi/tensorflow-gpu/1.5.0, 这个是下载1.5.0版本,只要更换版本号就可以下载其他版本了,eg.https://pypi.python.org/pypi/tensorflow-gpu/1.4.0。

      或者直接命令安装:

pip install --ignore-installed --upgrade tensorflow-gpu

  这里默认安装的是tensorflow_gpu-1.7.0-cp36m-win_amd64.whl,tensorflow_gpu-1.7.0要求cuda-9.0

5)验证安装是否成功

python
>>>import tensorflow as tf
>>>hello=tf.constant('Hello, Tensorflow')
>>>sess=tf.Session()
>>>print(sess.run(hello))
Hello, Tensorflow

 

 

 

 

 

参考文献:

1.http://blog.csdn.net/lucifer_zzq/article/details/76675239

posted on 2018-03-20 14:53  chamie  阅读(614)  评论(0编辑  收藏  举报