电脑睡眠之后程序停止，提示CUDA unknown error

报错内容

>>> print(torch.cuda.is_available())
/home/lab/anaconda3/envs/lzp_torch/lib/python3.10/site-packages/torch/cuda/__init__.py:124:
UserWarning: CUDA initialization: CUDA unknown error - this may be due
to an incorrectly set up environment, e.g. changing env variable
CUDA_VISIBLE_DEVICES after program start. Setting the available
devices to be zero. (Triggered internally at
/opt/conda/conda-bld/pytorch_1681197222248/work/c10/cuda/CUDAFunctions.cpp:108.)
  return torch._C._cuda_getDeviceCount() > 0
False
>>>

解决办法

sudo rmmod nvidia_uvm
sudo modprobe nvidia_uvm

参考：https://discuss.pytorch.org/t/userwarning-cuda-initialization-cuda-unknown-error-this-may-be-due-to-an-incorrectly-set-up-environment-e-g-changing-env-variable-cuda-visible-devices-after-program-start-setting-the-available-devices-to-be-zero/129335/4