python代码调用gpu/cuda异常,已解决

问题: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。

你可能感兴趣的:(python,pytorch,深度学习)