RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero. - Githubissues

LiuHC0428 / LAW-GPT

中文法律对话语言模型

1.06k stars 119 forks source link

RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero. #33

Open zhjc2014 opened 1 year ago

zhjc2014 commented 1 year ago

运行一段时间后出现上述错误

LiuHC0428 commented 1 year ago

应该是需要指定一个合理的设备，可以尝试一下os.environ['CUDA_VISIBLE_DEVICES']