RuntimeError:CUDA error:unknown error

运行A程序时，出现如下错误。
考虑原因：A程序在语料库ATIS中运行过，没有发生问题，但是之后运行了另一个程序B时，修改过gpu，如下所示：

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

如今在语料库Snips上运行A程序，却发生以下错误：
因此认为是GPU没有选择好，在A程序的main函数中添加一行程序后，顺利解决：

torch.cuda.current_device()