安裝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
文章來源:http://www.zghlxwxcb.cn/news/detail-592245.html
3.進入tensorflow官網查看對應的版本:
https://tensorflow.google.cn/install/source_windows?hl=zh-cn#cpu
添加鏈接描述文章來源地址http://www.zghlxwxcb.cn/news/detail-592245.html
到了這里,關于查看cudnn&cuda的版本以及對應的tensorflow版本的文章就介紹完了。如果您還想了解更多內容,請在右上角搜索TOY模板網以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網!