본문 바로가기

GPU3.2

Python 파이썬 셸 내에서 tensorflow가 GPU 가속을 사용하는지 확인하는 방법은 무엇인가요?, How to tell if tensorflow is using gpu acceleration from inside python shell? 질문 저는 우분투 16.04에서 두 번째 답변인 여기의 ubuntu의 기본 apt cuda 설치를 사용하여 tensorflow를 설치했습니다. 이제 제 질문은 tensorflow가 실제로 gpu를 사용하는지 어떻게 확인할 수 있는지입니다. 저는 gtx 960m gpu를 가지고 있습니다. import tensorflow를 실행하면 다음과 같은 출력이 나옵니다. I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcublas.so locally I tensorflow/stream_executor/dso_loader.cc:105] successfully opened CUDA library libcudnn.so l.. 2023. 12. 4.
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.