Hi @ptrblck!
I’ve managed to get it all working! The only issue I can is that I’m using Python with conda yet python seems to be using my pip install of PyTorch. For example, when I installed it via conda it found conflicts and wouldn’t detect my GPu at all (via torch.cuda.is_available()), but when I installed via pip. It worked and detect my GPU. A little bit strange to me!
Is it advisable to run PyTorch from conda rather than pip?
Thank you!