前言
最近使用Pytorch框架跑深度学习实验的过程,出现了以下错误
RuntimeError: cuda runtime error(999):unknown error at /pytorch/aten/src/THC/THCGeneral.cpp:50
解决方法
重新加载nvidia的内核模块,即可解决问题!
在命令行中输入以下代码
sudo rmmod nvidia_uvm
sudo modprobe nvidia_uvm
- rmmod(英文全拼:remove module)命令用于删除模块
- modprobe命令用于载入指定的个别模块