【内存占用不够】RuntimeError: CUDA out of memory.

当GPU内存占用不够时，将会出现RuntimeError: CUDA out of memory. 的错误，方法是将我们训练的模型转到相应的GPU上。

以便更好地训练我们的模型。

如果代码中已经写好了shell脚本则在脚本当中加入以下代码并修改：

CUDA_VISIBLE_DEVICES=3 python main.py --dir_data='dataset' \

其中的CUDA_VISIBLE_DEVICES=3代表你所占用的GPU显卡号在哪个显卡上面跑

如何查看哪个显卡上面还有内存：nvidia-smi指令查看