Hi I’m trying to get my model to run on the GPU.
I am using this example code: PyTorch Profiler With TensorBoard — PyTorch Tutorials 1.9.0+cu102 documentation
except in mine I call torch.cuda.is_available() to check it is available which returns True.
In Tensorboard however I can see it is running on CPU and my CPU usage goes up…
What am I missing here? I’m using a GTX 1080TI on windows 10. Thanks