Windows10上安装tensorflow-gpu和cuda

废话就不多说了,直接进入正题:

首先介绍我的电脑,戴尔燃7000系统是Windows10,显卡是MX150,这一款显卡并不能在哪个英伟达提供的支持cuda的显卡列表上找到,但是经过试验,发现它也可以支持cuda,后来我发现原来是英伟达的那个列表已经很久没有更新了,所以并不包含我的电脑的这一款显卡。

第一步:准备所需要的软件

anaconda、visual studio2015、CUDA9.0、CuDnn7.0

第二步:安装软件

尽可能的按照本文的安装顺序进行安装软件,否则可能会遇到一些问题

1.安装python

运行下载的anaconda就行了,这里建议使用anaconda安装python3.5,anaconda有很多优点能实现多个Python版本的共存这里需要安装的是Python3.5.安装完之后检查一下Python的版本如下:

Windows10上安装tensorflow-gpu和cuda

2.安装vs2015

这里要记住将VisualC++勾上就行了,因为需要vs2015编译一下文件,其他的默认就行了

3.安装CUDA

由于需要vs编译文件所以安装CUDA之前要确保vs2015已经安装成功。这里以一路默认好了。

安装好后的图如下:(图是借的)

Windows10上安装tensorflow-gpu和cuda

安装好后验证CUDA安装是否成功:

成功如下图:

Windows10上安装tensorflow-gpu和cuda

4.使用vs编译测试文件

进入C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0,使用vs2015打开Samples_vs2015

Windows10上安装tensorflow-gpu和cuda

选择编译1_Utilities中的所有文件,在1_Utilities上右键选择生成就行,在这里要注意红框部分的debug改成release

Windows10上安装tensorflow-gpu和cuda

编译成功后如下图:

Windows10上安装tensorflow-gpu和cuda


成功之后在C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\bin\win64\Release会多出如下文件

Windows10上安装tensorflow-gpu和cuda

这里我们验证一下deviceQuery和bandwidthTest,具体就是在cmd中运行这两个文件

成功后如下图:

Windows10上安装tensorflow-gpu和cuda

还有一个:

Windows10上安装tensorflow-gpu和cuda

这两张图出现Result= PASS就说明成功了。

5.安装CuDnn

解压缩下载的CuDnn文件,会得到bin、include、lib这三个文件夹,将这三个文件夹复制到C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.0下

Windows10上安装tensorflow-gpu和cuda

6.检验系统环境变量

在安装CUDA是如果一直是默认的情况下那么在环境变量中就已经存在CUDA_PATH和CUDA_PATH_V9_0,我们还要手动将C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.0\bin加入path中

Windows10上安装tensorflow-gpu和cuda

7.安装GPU版本的tensorflow

打开anaconda的命令行输入pip install tensorflow-gpu(这里要注意的是要先安装pip和注意安装GPU版本的tensorflow)

安装好后检查一下是否安装成功:

Windows10上安装tensorflow-gpu和cuda

8.检验是否成功

Windows10上安装tensorflow-gpu和cuda

可以发现是成功了的。