安装tensorflow-GPU时,无法确定自己电脑需要安装哪个版本,这是在官网查询对应的版本即可
1.查看cuda的版本:
Win+R输入cmd——nvcc -V
输入"nvcc -V"
可以看到cuda的版本为10.2
2.查看cudnn的版本:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include
在此路径下找到“cudnn.h
”文件
右键,选择以记事本
打开
可以看到cudnn的版本为7.6.5
文章来源:https://www.toymoban.com/news/detail-592245.html
3.进入tensorflow官网查看对应的版本:
https://tensorflow.google.cn/install/source_windows?hl=zh-cn#cpu
添加链接描述
文章来源地址https://www.toymoban.com/news/detail-592245.html
到了这里,关于查看cudnn&cuda的版本以及对应的tensorflow版本的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!