linux调用gpu跑模型时遇到问题

Question

环境配置都完成了，运行时出来了如下问题

1.之前试过将cudnn从7.65换为推荐的7.6版本没有效果
2.显卡大小是40g不知道是不是大小不够？跑的时候也用命令查过利用率也没到100%
3.显卡的最大兼容版本是12.2 需不需要下载12.2或者就近的版本？（之前就是因为有bug从12.2降到10.1的）

我的环境:
nvidia smi 535.129.03
cuda 10.1.243
cudnn 7.6
python3.7
tensorflow-gpu 2.2.0

Answer 1

有没有可能是TensorFlow 的版本 和 cuda 版本不匹配？ TensorFlow兼容性问题就是多，不如换pytorch