运行A程序时,出现如下错误。
考虑原因:A程序在语料库ATIS中运行过,没有发生问题,但是之后运行了另一个程序B时,修改过gpu,如下所示:
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
如今在语料库Snips上运行A程序,却发生以下错误:
因此认为是GPU没有选择好,在A程序的main函数中添加一行程序后,顺利解决:
torch.cuda.current_device()
运行A程序时,出现如下错误。
考虑原因:A程序在语料库ATIS中运行过,没有发生问题,但是之后运行了另一个程序B时,修改过gpu,如下所示:
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
如今在语料库Snips上运行A程序,却发生以下错误:
因此认为是GPU没有选择好,在A程序的main函数中添加一行程序后,顺利解决:
torch.cuda.current_device()