https://videosashka.com/watch/uCM9NgSBHi4/python-check-if-gpu-available-pytorch