Cuda runtime error (999)

本人使用的是Ubuntu18.04，使用GPU进行深度炼丹的时候出现了Cuda runtime error(999)的错误。

参考：https://discuss.pytorch.org/t/cuda-runtime-error-999/69658/11

“ Ubuntu seems to have some issues with sleep/suspend (or maybe Linux in general?).”

解决办法为在终端中使用下列命令：

sudo rmmod nvidia_uvm
sudo modprobe nvidia_uvm