win10(NVIDIA MX150) 安装tensorflow-gpu及keras

时间:2024-03-26 17:47:08

折腾了一天多TensorFlow-GPU的坑,安装了两次cuda。

本篇基于NVIDIA GeForce MX150 安装 CUDA,cuDNN,Python(anaconda)TensorFlow-GPU(Windows10操作系统)

 

安装顺序: 查看配置环境 —> CUDA Toolkit —> cuDNN —> Visual Studio 2015 Community —> anaconda虚拟环境 —> TensorFlow-GPU —> 测试

 

一、安装环境

我的电脑配置如下:

1.windows 10 64bit 专业版

2.Anaconda 2018.12 Python3.7(需要降级为python3.6)

3. Visual Studio 2015 Community

4.显卡支持NVIDIA GeForce MX150

5. cuda_9.0.176_win10_network.exe

6. cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0

7.TensorFlow-gpu 1.7.0

8.Keras 2.2.4

9.numpy 1.16.2

10.PyQt5 5.12.1

 

软件版本说明:

1.Visual studio 2015: 这个版本和CUDA8.0和CUDA9.0都兼容,VS2017和CUDA8.0不兼容,与CUDA9.0兼容。

2.Tensorflow-GPU版本坑

官方tensorflow1.6\tensorflow1.7版本均不支持cuda9.1。所以如果要安装cuda9.1下的tensorflow1.6或1.7需要使用网上一个大神编译的版本。由于我最终更换为cuda9.0了就没有试大神的版本。

网上看很多人都是卸载cuda9.1无法卸载干净,所以我就没卸载,抱着试试看的心态,直接安装了cuda9.0。然后把我的电脑环境变量中带有cuda9.1字样的都删除了

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

 

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

二、安装Visual Studio 2015 Community

安装cuda前要先安装visual Studio,否则无法使用

先上Visual Studio 2015 Community资源百度网盘 提取码:f1qz (对!就是那么贴心)

 

安装VS2015前,请先断网,不然安装过程会下载一堆东西,过程极慢。安装时只选择Visual C++部分,其他都可以不装,这样安装起来更快。安装要一会儿,请耐心等待。

 

安装过程可参考博客

需要注意的是:

安装路径可以*选择

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

自定义安装不需要 VS2015 的全部组件,只需要与 C/C++ 相关的组件,所以这里只选择了“Visual C++”,将其它用不到的组件全部取消勾选。

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

然后。。。请开始你漫长的安装等待吧。。。(喝杯茶,看个啥)

 

三、安装CUDA Toolkit

需要使用TensorFlow-GPU请先查看电脑是否支持GPU显卡:

1.设备管理器 —> NVIDIA GeForce MX150

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

如果显卡不是NVIDIA GeForce MX150 请查看 网站或下图配置你的CUDA

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

2.查看NVIDIA GeForce MX150 所适合的CUDA版本:

控制面板 —> NVIDIA控制面板 —> 系统信息 —> 组件

确定NVIDIA GeForce MX150支持CUDA9.1版本

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

从CUDA官网(https://developer.nvidia.com/cuda-toolkit-archive)下载cuda_9.0.176_win10_network.exe 如下图

 

https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocalwin10(NVIDIA MX150) 安装tensorflow-gpu及keras

 

安装CUDA:

  1. 文件解压

这个地方的地址默认就好,应该就是暂时放一下,直接点OK

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

 

b.系统硬件兼容性

这个地方我的笔记本系统兼容性检测说无法识别硬件,是否要继续!”This graphics driver could not find compatible graphics hardware...“,这个地方可能是“因为笔记本厂商一般都给自己的机器采用的显卡添加了独特的硬件ID,导致CUDA没有包含这个ID,驱动认不出显卡”,但是也不用担心,可以继续往下安装

 

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

 

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

c.安装选择高级

勾选内容部分我是把所有已经安装的驱动的勾点掉了,只安装没有的驱动,安装路径默认了,在这个地方可以记录一下自己安装的文件路径

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

更新CUDA、Driver(不更新的话有可能导致NVIDIA控制面板打不开)、physx

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

 

安装位置不需要改变,后面过程直接下一步即可。

 

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

安装后可进入cmd里面输入nvcc -V 查看是否安装成功,如下所示即安装成功

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

 

四、安装cuDNN

cuDNN可以在前面GPU的基础上在提升1.5倍的速度,由NVIDIA开发,因此可以在官网中下载。在官网(https://developer.nvidia.com/rdp/cudnn-archive)下载需要先注册。下载cuDNN v7.1.4 (May16, 2018), for CUDA 9.0 Windows10版本 (注意一定要与CUDA版本号对应)

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

 

下载完毕解压会看到3个文件夹如下:

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

下载完成后解压缩。里面有bin、include、lib三个目录,将三个文件夹内容复制到安装CUDA的地方覆盖对应文件夹内容,CUDA默认文件夹在:

 

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

1

检查一下环境变量 我的电脑—> 属性—> 高级系统设置—> 环境变量—>系统变量

中加入CUDA 的bin所在目录环境

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

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

如何cuDNN验证是否安装成功?

cuDNN无法直接验证安装成功,在之后TensorFlow-GPU安装完毕后

import tensorflow as tf 若报错ImportError: Could not find 'cudnn64_7.dll'.则证明安装失败。

 

五、配置Anaconda虚拟环境

在Anaconda中配置虚拟环境有两种方法:

1.可以直接打开Anaconda Navigator,创建TensorFlow–GPU虚拟环境

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

2.也可以打开Anaconda Prompt ,输入下面命令用于创建一个虚拟环境,名字叫TensorFlow–GPU, 同时指定python的版本,如果本机内没有安装这个版本的python,就会自动下载安装.

 

conda create -n TensorFlow--GPU python=3.6

 

配置完虚拟环境env后可在Anaconda Prompt 中输入:

**环境

activate TensorFlow--GPU

 

退出环境

deactivate

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

 

在虚拟环境中安装Spyder和Jupyter,此处我已安装Spyder

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

在Environments中同样安装Spyder与ipython 以防报错,此处我已安装:

 

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

注意: 如果Anaconda默认安装的python版本是3.7 应该降到python 3.6版本,打开Anaconda Prompt在TensorFlow–GPU 虚拟环境中输入:

 

conda install python=3.6

1

六、安装TensorFlow-GPU

打开Anaconda Prompt ,**环境后,安装TensorFlow-gpu即可(太慢也可用镜像安装)。

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

注意:TensorFlow-GPU版本使用1.6

 

七、测试

在TensorFlow-GPU虚拟环境中打开python 测试一下安装是否成功。

输入:

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

等待片刻,如下图所示,即安装成功:

 

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

也可以跑个tensorflow语句试试试

 

import tensorflow as tf   # 引入 Tensorflow 库

gjm = tf.constant("Hello World ! I love TensorFlow ! ")   #创建一个常量 Operation (操作)

sess = tf.Session()   #启动一个 TensorFlow 的 Session(会话)

print (sess.run(gjm))   #运行 Graph (计算图)  python3.x 专用语法

sess.close()   #关闭 Session (会话)

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

在虚拟环境下的Spyder测试一下:

win10(NVIDIA MX150) 安装tensorflow-gpu及keras

 

七、安装keras

Pip install keras==2.2.4

 

八、报错经验

1.由于创建了虚拟环境,各类python包需要重新安装,注意安装的版本号对应(一、安装环境 中几个主要的包及版本已进行说明)

2.如果cuDNN版本与CUDA的版本不匹配import TensorFlow时会报错:

ImportError: Could not find 'cudnn64_7.dll'.

需要仔细核对cuDNN 与CUDA的版本匹配。