相信许多朋友在使用深度学习时,将模型放在GPU
上运行,然后就会报一堆跟CUDA
相关的错误,于是开始百度这些错误该怎么修复。得到的结果可能大多数都是该怎么升级CUDA
之类的。当按照网上的流程一步步走下来后,却始终无法解决这些问题。
而实际上在我遇到的所有跟CUDA
相关的错误,其实都与CUDA
没有任何关系,比如有可能是你的Embedding
层的维度与词表不匹配等。发生这些错误的时候,我建议先将模型放到CPU上运行,看会不会发生错误,再决定你该如何修改代码。
相信许多朋友在使用深度学习时,将模型放在GPU
上运行,然后就会报一堆跟CUDA
相关的错误,于是开始百度这些错误该怎么修复。得到的结果可能大多数都是该怎么升级CUDA
之类的。当按照网上的流程一步步走下来后,却始终无法解决这些问题。
而实际上在我遇到的所有跟CUDA
相关的错误,其实都与CUDA
没有任何关系,比如有可能是你的Embedding
层的维度与词表不匹配等。发生这些错误的时候,我建议先将模型放到CPU上运行,看会不会发生错误,再决定你该如何修改代码。