I believe that when many friends use deep learning, they put the model GPU
on and run it, and then a bunch of related errors will be reported CUDA
, so they started to Baidu how to fix these errors. Most of the results obtained may be how to upgrade CUDA
and the like. After following the online process step by step, these problems were still unable to be solved.
In fact, all CUDA
the related errors I encountered have CUDA
nothing to do with it. For example, it may be that the dimensions of your Embedding
layer do not match the vocabulary, etc. When these errors occur, I suggest running the model on the CPU first to see if any errors occur, and then decide how to modify the code .