(租用服务器GPU大概率是这样)RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
在服务器上租用实例(创建的时候自带的框架以及pytorch),跑模型的时候出现的这样的错误。当时租用的是3090-24G,pytorch1.8.1,cuda 11.1,python3.8,ubuntu 18.04。自己刚入门,对版本这些没什么要求,租用实例的时候心想肯定时越高越好(之前听说最新的版本会出现一些问题,相当于测试版本),就租用了稍微新一点的版本。第一次跑模型使用上面自带的环境,结果没有原论文里面的好,还差了一大截。经过github的issue,解决方案是pytorch的版本会影响模型指标。
原创
2023-03-06 11:34:13 ·
1641 阅读 ·
0 评论