很多时候这个错误都以为是Cuda安装的问题。其实并非全部如此,有一部分原因是你的GPU显存被占用光了,再运行就出错了。
我就遇到过,rnn(embedded),使用LSTM/GRU处理Embedding后的序列,为何会出现这个问题。
建议nvidia-smi查询一下GPU的使用情况,Kill -9 掉部分程序。尤其是使用Jupyter Lab运行的,没有即使释放显存,会出现这个问题。
很多时候这个错误都以为是Cuda安装的问题。其实并非全部如此,有一部分原因是你的GPU显存被占用光了,再运行就出错了。
我就遇到过,rnn(embedded),使用LSTM/GRU处理Embedding后的序列,为何会出现这个问题。
建议nvidia-smi查询一下GPU的使用情况,Kill -9 掉部分程序。尤其是使用Jupyter Lab运行的,没有即使释放显存,会出现这个问题。