PyTorch2.1 Python PyTorch가 GPU를 사용하는지 어떻게 확인할 수 있나요?, How do I check if PyTorch is using the GPU? 질문 PyTorch가 GPU를 사용하고 있는지 확인하는 방법은 무엇인가요? nvidia-smi 명령어는 GPU 활동을 감지할 수 있지만, 저는 파이썬 스크립트 내부에서 직접 확인하고 싶습니다. 답변 이러한 함수들이 도움이 될 것입니다: >>> import torch >>> torch.cuda.is_available() True >>> torch.cuda.device_count() 1 >>> torch.cuda.current_device() 0 >>> torch.cuda.device(0) >>> torch.cuda.get_device_name(0) 'GeForce GTX 950M' 이는 다음을 알려줍니다: CUDA가 사용 가능하며 한 장치에서 사용할 수 있습니다. Device 0은 GPU GeForce G.. 2023. 10. 11. 이전 1 다음
