win10 安装 GPU版TensorFlow

WIN10安装CUDA9+CUDNN+TENSORFLOW1.9.0

安装环境:笔记本电脑win10专业版操作系统 + 1050显卡

1、安装CUDA9

地址:https://developer.nvidia.com/cuda-90-download-archive
在这里插入图片描述

安装时会先选择解压位置作为临时存放,可以自己定义。

安装过程中可能会弹出以下警告:
在这里插入图片描述

这个是由于安装的显卡驱动版本高于CUDA9所自带的驱动版本号,点击继续即可。安装的时候建议选择自定义,不选VS的组件:
在这里插入图片描述

自己可以选择路径安装,结束。

2、下载安装CUNNN

下载cudnn,下载地址:https://developer.nvidia.com/cudnn,进去之后,找到对应的版本,点击Download,他会要你注册登录,然后填个问卷,才可以下载,这一步自行按照网站提示做就可以了。提交完问卷后,就可以点击下载了。

选择以下的版本均可。

在这里插入图片描述

下载完成后解压,把解压后的3个文件夹放入CUDA9安装对应的文件夹中。
在这里插入图片描述

这里我的CUDA9路径是自己定义的。
在这里插入图片描述

3、安装tensorflow-gpu V1.9.0

tensorflow-gpu 版本过高运行程序时,会报错。实测V1.11.0也是可以正常使用的。

pip install --index-url https://pypi.douban.com/simple tensorflow-gpu==1.9.0

4.检测是否成功

打开任意一个python编辑环境,输入以下代码

import tensorflow as tf

hello = tf.constant(‘hello,TensorFlow!’)

sess = tf.Session()

print(sess.run(hello))

最后一行显示 b’hello,TensorFlow!'说明Tensorflow配置成功!!

注意:这里安装的是TensorFlow1.x版本,若要安装2.x版本,方法相同,只是换为CUDA10+CUDNN+TensorFlow2.0

你可能感兴趣的:(java,后端,tensorflow,深度学习,机器学习)