0
点赞
收藏
分享

微信扫一扫

rtx gpu kera loss不降或为nan

爪哇驿站 2022-03-21 阅读 157

更换tensorflow版本为2.4.0以上,好像说是30显卡的问题(我是rtx3060)
如:
cudatoolkit=11.3 cudnn=8.2.1 tensorflow-gpu=2.5.0

安装(国内源好下载的版本):
conda install tensorflow-gpu=2.5.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

tensorflow cuda cudnn对应版本 参考:
https://tensorflow.google.cn/install/source_windows

举报

相关推荐

0 条评论