本人使用的是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