Hello, I’m not sure if you’ve read the same commit, there is a line I pasted above, here is a screenshot of it:

image2461×1651 318 KB

there is a line 300 with code for setting up device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

If this is what you’re asking about