问题:RuntimeError: CUDA error: kernel launch failure when calling `cusparseXcoo2csr(handle, coorowind, i_nnz, i_m, csrrowptr, CUSPARSE_INDEX_BASE_ZERO)`
原因:大概率的原因是:pytorch版本与服务器上的cuda版本不匹配导致的;
还有一部分原因可能是gpu都被占用了,没有资源了,需要kill掉一些进程。
解决:首先查看自己服务器上的cuda版本,使用命令:nvidia-smi;
然后在pytorch官网下载对应cuda版本的pytorch,地址:Previous PyTorch Versions | PyTorch。