tensorflow 不调用显卡 win10 安装GPU版本te
2018-08-21 本文已影响4人
壹刀_文
换了一个好显卡,训练模型的时候比AWS租的慢几十倍??
看看这一轮训练的时间,长到天荒地老。
![](https://img.haomeiwen.com/i12043887/fd41d838564452ad.png)
这简直是一件不能接受的事,难道钱打水漂了?
No,我们 too young too simple!
先使用 nvidia-smi 命令检查一下。
'nvidia-smi' 命令可以在linux下直接使用,在windows下直接执行会报错:
![](https://img.haomeiwen.com/i12043887/a450d113573da7e6.png)
直接解决办法是到 'NVSMI' 目录下去执行,通常目录位置为:
C:\Program Files\NVIDIA Corporation\NVSMI
执行结果如下:
![](https://img.haomeiwen.com/i12043887/1b43eba0bb479c99.png)
看吧, Process name里面没有一个python, 说明根本没有调用显卡来进行计算。
- 先卸载目前的tensorlfow :
pip uninstall tensorflow -y - 然后安装GPU版本的tensorflow:
pip install tensorflow-gpu -UI
如果pip下载实在是太慢,或者容易掉线的话,建议直接复制地址用工具下载,速度杠杠
下载whl文件,安装只需要到文件所在位置,执行
pip install tensorflow_gpu-1.10.0-cp36-cp36m-win_amd64.whl
安装结束! 运行试一下,提示错误就来了-_-!!~:
![](https://img.haomeiwen.com/i12043887/9b136144a2306c0c.png)
去下载了windows版本
![](https://img.haomeiwen.com/i12043887/b6aa6b2ecf9bddc0.png)
下载bese installer 以及 4个 patches
在这之前要记得安装NVIDIA驱动,地址为:
https://www.nvidia.cn/Download/index.aspx?lang=cn
以上成功安装完成后 , 不出意外的还有问题(喝口水平静一下)
![](https://img.haomeiwen.com/i12043887/79591d237a94d064.png)
下载地址:
https://developer.nvidia.com/cudnn
这个下载需要注册,验证邮件以及填一下表格。
下载后得到三个文件夹 ,需要复制到路径中。
![](https://img.haomeiwen.com/i12043887/74f0751fc9c351b3.png)
查找路径可以输入命令:
echo %path%
![](https://img.haomeiwen.com/i12043887/bcfdaa934a6df555.png)
找到本机的CUDA地址,我的是
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
三个文件夹复制过去。再调用一下
![](https://img.haomeiwen.com/i12043887/e083389a1535be09.png)
表示------‘呵呵‘
升级 numpy / 卸载 unmpy 再重装:
pip uninstall numpy
pip install numpy --upgrade
numpy 的版本需要注意一下
![](https://img.haomeiwen.com/i12043887/34d577a83e3bb868.png)
需要 numpy<=1.14.5,>=1.13.3,
再运行 nvidia-smi 看看
![](https://img.haomeiwen.com/i12043887/b8ad6598853d7fa0.png)
运行速度改变了
![](https://img.haomeiwen.com/i12043887/f4ec60a37abdcf5c.png)