python5.6 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. Python 파이썬 진행 바, Python Progress Bar 질문 나의 스크립트가 시간이 걸릴 것으로 예상되는 작업을 수행할 때 진행 막대(progress bar)를 어떻게 사용하나요? 예를 들어, 완료까지 시간이 걸리는 함수가 있고, 완료되면 True를 반환합니다. 이 함수가 실행되는 동안 진행 막대를 표시하는 방법이 있을까요? 실시간으로 표시해야 하므로 어떻게 해야 할지 모르겠습니다. 이를 위해 thread가 필요한가요? 잘 모르겠습니다. 현재 함수가 실행되는 동안 아무것도 출력하지 않고 있지만, 진행 막대가 있으면 좋을 것 같습니다. 또한 코드 관점에서 이를 어떻게 할 수 있는지에 더 관심이 있습니다. 답변 tqdm을 사용하면(conda install tqdm 또는 pip install tqdm), 루프에 진행 막대를 쉽게 추가할 수 있습니다: from ti.. 2023. 10. 7. 이전 1 2 다음