How do I check if pytorch is using the GPU? It’s possible to detect with nvidia-smi if there is any activity from the GPU during the process, but I want something written in a python script.
-
1010 Avenue of the Moon
New York, NY 10018 US. -
Mon - Sat 8.00 - 18.00
Sunday: Closed - Get A Quote