win10安装tensorflow gpu版本

一、安装cuda

查看电脑支持的cuda版本

安装cuda（相当于运算平台）才能使得GPU用于我们所需的复杂的计算

首先打开控制面板，然后搜索“nvidia 控制面板”，打开

来到这个界面，点击帮助中的系统信息

点击组件后直接查看cuda版本号，我的是9.1

下载安装cuda 

然后按照这个版本去下：

9.0版本： https://developer.nvidia.com/cuda-90-download-archive

9.1版本： https://developer.nvidia.com/cuda-91-download-archive

10.0版本： https://developer.nvidia.com/cuda-downloads

进去后是这个界面，有联网安装版本和不联网安装版本

不联网的1.4G，慢慢下吧…也可以选择联网的，都挺慢的。

安装的时候最好默认吧，如果版本不适合是要重新安装其他版本的。

 

二、安装cudnn

cudnn是用来加速GPU计算的

官网有下载方式，但是需要登陆注册，如果觉得很麻烦或者是很慢可以考虑搜一下百度云。

官网下载地址： https://developer.nvidia.com/rdp/cudnn-download

注意要根据自己的版本下载，我下载的是for cuda 9.2

安装吧，100+m好像，下载后是一个压缩包，解压到你指定文件夹。

然后把这个文件夹下面的bin路径添加到path环境变量里面（控制面板->系统和安全->系统->高级系统设置->环境变量->系统变量->找到path变量，点击进去->新建然后写上刚才的bin目录->全部确定）

 

三、安装tensorflow-gpu

pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

打开cmd，运行上面这句。这里换成清华源，更快。然后就可以运行程序了。

你可以打开C:\Program Files\NVIDIA Corporation\NVSMI，把下面这个拖进cmd，查看现在显卡在跑的程序和占用率以及你的显卡类型任务PID。

 

 

不得不说gpu版本事真多…如果CPU版本简单一句话搞定…

pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple