How to enable gpu rtx 3060 6gb on system for executing code on jupyter notebook?

I want to use the gpu of my system(RTX 3060 6GB) to execute the Jupyter notebook for training the model using Pytorch. But I am unable to execute the code using gpu. It is showing the device as cpu not gpu.


Welcome to the NVIDIA Developer forums. You posted in the Omniverse category, your question might be better suited for the AI & Data Science section. I found this post that might offer a solution.

Tom K