解决RuntimeError: CUDA error: out of memory

跑代码时出现这个错误，输入以下命令

$ nvidia-smi -l

原来是卡0内存爆了，于是指定相对空闲的2卡

CUDA_VISIBLE_DEVICES=2 python my_script.py

可以正常运行

原文：https://www.cnblogs.com/Tony100K/p/14520567.html