win7/win10下cuda8.0安装跑gpu版tensorflow

时间:2024-04-03 13:47:21

简单写下配置过程吧,挺简单的:

0.win7/10 X64系统

1.安装vs2013

2.安装Anaconda3(需要里面的python环境)

3.下载CUDA8.0(https://developer.nvidia.com/cuda-downloads),下面两个exe文件都下载

win7/win10下cuda8.0安装跑gpu版tensorflow

           补充说明:我电脑里原来是cuda7.5,故需要先卸载干净,我是这么做的,有需要的可以借鉴:

                 (1)把下图中的几项用电脑管家全部卸载干净

             win7/win10下cuda8.0安装跑gpu版tensorflow

                 (2)删除C:\Program Files\NVIDIA GPU Computing Toolkit 文件夹

                     删除 C:\ProgramData\NVIDIA GPU Computing Toolkit 文件夹

                     删除C:\ProgramData\NVIDIA Corporation\CUDA Samples 文件夹

4.运行exe

       win7/win10下cuda8.0安装跑gpu版tensorflow

                 默认下一步到最后

5.再运行exe

          win7/win10下cuda8.0安装跑gpu版tensorflow

                默认下一步到最后

6.下载cuddn5.1/cudnn6.0(如果想要了解cuddn和cuda的区别可以看该博客http://blog.csdn.net/fangjin_kl/article/details/53906874

7.解压cuddn5.1/cudnn6.0,把如下的三个文件夹替换到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0夹下(该目录是我的cuda8.0的安装目录)

win7/win10下cuda8.0安装跑gpu版tensorflow

 这里需要在系统变量里设置下面几个变量:

   CUDA_PATH:          C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

   CUDA_BIN_PATH: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

   CUDA_LIB_PATH:  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

   CUDA_PATH_V8_0: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

   CUDA_SDK_BIN_PATH: C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\bin\win64

   CUDA_SDK_LIB_PATH: C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\common\lib\x64

   CUDA_SDK_PATH:         C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0

win7/win10下cuda8.0安装跑gpu版tensorflow

8.查看cuda8.0是否安装成功,可以在CMD窗口下敲指令nvcc -V

win7/win10下cuda8.0安装跑gpu版tensorflow

9.再运行一个Sample例子 打开C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0\1_Utilities\deviceQuery解决方案,编译下,出来的结果为

win7/win10下cuda8.0安装跑gpu版tensorflow

10.pip安装tensorflow gpu库(http://blog.csdn.net/u014365862/article/details/53868578)

         在Anaconda Prompt里输入(不是cmd跳出的窗口内输入) pip install tensorflow-gpu

      win7/win10下cuda8.0安装跑gpu版tensorflow

11.安装完后输入import tensorflow试试

win7/win10下cuda8.0安装跑gpu版tensorflow

其实win7,win10的安装都一个样,关键是要文件下载对,然后还有一个注意的地方就是设置环境变量的时候不要弄错了,默认cuda安装的路径,就直接复制粘贴上面的路径就好了。