jianchang512/clone-voice

BUG: 判断是否有CUDA错误

thunderyan opened this issue · 1 comments

thunderyan commented

cfg.py 中，
device = "cuda" if os.getenv('CUDA','')=='CUDA' and torch.cuda.is_available() else "cpu"

应该改为：
device = "cuda" if os.getenv('DEVICE','')=='CUDA' and torch.cuda.is_available() else "cpu"

iShot_2024-03-07_12 57 48

jianchang512 commented

ok,多谢反馈