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