问题描述
我想在GPU上训练RetinaNet，但是在训练过程中出现错误：
Failed to get convolution algorithm. This is probably because cuDNN Failed to initialize
我必须具有tensorflow == 2.2或tensorflow == 2.1和keras == 2.3.1的版本。我已经为CUDA 10.1安装了CUDA 10.1，CUDA Toolkit 10.1 update2和cuDNN v7.6.5。我的python版本== 3.7。
然后，我创建一个虚拟环境，并安装所有这些版本的tf，keras和python。
您知道哪里可能有错误吗？
解决方法
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）