RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)

在写pytorch情感分类时，代码出现这个问题

经过一番查找发现，我的代码出错的地方如下

如果这个nn.Embedding()里面的vocab_size写的不正确，比如你对于处理文本数据使用的词典数是1000，但是你在这里写的小于1000，在程序运行时，它在这句代码下面的执行过程中报错，实际还是这个的错误。
我还发现，当我们这里写错时，在CPU上运行和在GPU上运行会显示不一样的错误。

一.在CPU上运行上面代码时，报错是IndexError: index out of range in self

二.在GPU上运行上面代码时,报错是RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)

希望这次回答能够对您有所帮助!