EamesLionel

I tried manually adjusting all `'cuda:0'` into `'cuda:0' if torch.cuda.is_available() else "cpu"`, and it worked.

可以，记得把`"cuda:0"`改成`"cuda" if torch.cuda.is_available() else "cpu"`，能跑，效果也还不错的