编程速记：CUDA OOM处理-CUDA GPU 隔离

Problem：CUDA OOM

解决方案

一般来说，Pytorch默认使用的是CUDA 0
可以指定显卡

CUDA_VISIBLE_DEVICES=1 python xxx.py

当然对于上面这种方案的另一种做法，先设置环境变量再启动

对于LINUX环境

设置临时环境变量

export CUDA_VISIBLE_DEVICES=1

设置永久环境变量
step1.打开~/.bashrc
step2.在最后一行添加上export CUDA_VISIBLE_DEVICES=1 （注意：如果要设置多个GPU块可以使用逗号隔开编码）
step3.source ~/.bashrc

之后再在命令行执行python文件