安装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
**
3.进入tensorflow官网查看对应的版本:
https://tensorflow.google.cn/install/source_windows?hl=zh-cn#cpu
添加链接描述
本文转载自: https://blog.csdn.net/qq_44961737/article/details/129516903
版权归原作者 MADAO 所有, 如有侵权,请联系我们删除。
版权归原作者 MADAO 所有, 如有侵权,请联系我们删除。