失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > Window10 RTX2070显卡+tensorflow_gpu1.14+cuda10.0+cudnn+7.4.2的环境配置

Window10 RTX2070显卡+tensorflow_gpu1.14+cuda10.0+cudnn+7.4.2的环境配置

时间:2023-09-20 15:36:16

相关推荐

Window10 RTX2070显卡+tensorflow_gpu1.14+cuda10.0+cudnn+7.4.2的环境配置

1.判断目前机器的显卡驱动是否为最新,是否需要更新驱动。

2.根据要安装的tensorflow_gpu的版本,注意最开始安装的tensorflow 1.14版本要求cuda的版本是10.0,而安装时10.1,所以报错。

现在NVIDIA的显卡驱动程序已经更新到 10.1版本,最新的支持CUDA 10.1版本的cuDNN为7.5.0

3.下载cudnn 对应的包,放入到cuda安装的文件夹中。

4。在cmd下输入 nvcc -V判断对应的cuda版本是否安装正常。

5.修改pip下载地址,从清华镜像下载tensorflow。

6.测试tensorflow——gpu环境是否配置好

1 #!D:/Code/python 2 # -*- coding: utf-8 -*- 3 # @Time : /8/26 18:28 4 # @Author : Johnye 5 # @Site : 6 # @File : test_tensorflow_gpu.py 7 # @Software: PyCharm 8 import tensorflow as tf 9 hello=tf.constant("hello111")10 sess = tf.Session()11 print(sess.run(hello))

7.将numpy降级为1.16,以下警告解决

如果觉得《Window10 RTX2070显卡+tensorflow_gpu1.14+cuda10.0+cudnn+7.4.2的环境配置》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。