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