win10下安装tensorflow-gpu
步揍1:
首先电脑要安装独立显卡
步揍2:
下载cuda (它就相当于GPU的驱动)和对应的本版的cudnn
这里我的cuda下载的是 ,下载地址https://developer.nvidia.com/cuda-downloads
下载过后,这里我们默认安装就好,安装的主目录一般在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,
进去之后我们需要把主目录下的bin和 lib/x64 ,这两个文件夹添加到环境变量里。
加载之后,这个CUDA就生效了。
步揍3:CUDA 安装过以后要安装cudnn,它是给深度学习计算加速的。
3.1.cuDNN下载(要先注册个账号才能下载)
https://developer.nvidia.com/rdp/cudnn-download
这里我下载的版本是当时最新的5.1版
3.2.解压压缩包,把压缩包中bin,include,lib中的文件
分别拷贝到C:\Program Files\NVIDIA GPU Computing
Toolkit\CUDA\v8.0目录下对应目录中
3.3把C:\Program Files\NVIDIA GPU Computing
Toolkit\CUDA\v8.0\extras\CUPTI\libx64\cupti64_80.dll
拷贝到C:\Program Files\NVIDIA GPU Computing
Toolkit\CUDA\v8.0\bin
4.安装tensorflow
用管理员身份打开 cmd 输入 pip install tensorflow-gpu
注意:如果安装出现下面错误的话 ,可能是缺少的运行库(Visual C++ 2015 redistributable (x64
version).),到下面下载就可以
https://www.microsoft.com/en-us/download/details.aspx?id
上面方法如果还是安装不了的话,用下面链接方法,同时下面的方法,还告诉你怎么创建虚拟环境
https://blog.csdn.net/weixin_39290638/article/details/80045236