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

AI & Data Science Deep Learning (Training & Inference) cuDNN
,
1

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.

2

Hello,

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.

Best,
Tom K