Hi @AnderBiguri
Our last merge seems to have been an oversight by both of us: whenever we have torch.cuda.<any_method>() the initialization demands GPU support and hence still throws the same error on my laptop. So I believe we should change it back to my original proposal, device = torch.device('cuda' if torch.cuda.is_available() else 'cpu'). I am currently doing my bachelor's thesis in CT reconstruction and would like to prototype some torch models on my laptop before moving them to the GPU server in my uni. Thanks!
Hi @AnderBiguri
Our last merge seems to have been an oversight by both of us: whenever we have
torch.cuda.<any_method>()the initialization demands GPU support and hence still throws the same error on my laptop. So I believe we should change it back to my original proposal,device = torch.device('cuda' if torch.cuda.is_available() else 'cpu'). I am currently doing my bachelor's thesis in CT reconstruction and would like to prototype some torch models on my laptop before moving them to the GPU server in my uni. Thanks!