解决RuntimeError: CUDA error: out of memory

问题描述：

在使用 PyTorch 中加载模型时，出现 CUDA 错误: out of memory。主要有以下三个可能原因：

RuntimeError: CUDA error: out of memory

原因分析：

1. CUDA 显存不足：可能是因为当前显卡显存不足。
   解决办法：考虑更换显存更大的显卡。

2. 显卡被占用：当存在多个显卡时，默认使用 0 号显卡。即使 1 号显卡空闲，但如果 0 号显卡被占用，仍然会导致报错。
   解决办法：可以让其他显卡不可见，从而使 0 号显卡变为空余的 1 号显卡。