목록Device (1)

areum

[Tensorflow] 텐서플로우 GPU이용하여 실행하도록 하는 방법

tensorflow 를 이용하여 머신러닝을 실행할때, cpu나 gpu를 사용하게 된다. gpu를 시용하면 cpu만 사용하는 것에 비해 속도가 약 10배 이상 차이가 난다. 그렇기에 모델을 학습시킬때 gpu를 사용하는 것이 훨씬 효율적이며, cpu에 부하도 훨씬 적을 것입니다. 아래에서 gpu로 실행하도록 설정하는 두가지 방법에 대해서 알아보겠습니다. 아래 방법 사용 전 gpu device가 잡히는지 확인해야 합니다. from tensorflow.python.client import device_lib print(device_lib.list_local_devices()) 위 코드에서 gpu가 잡히는게 확인 되었다면 아래 방법을 실행하면 됩니다. 방법 1: 전체 GPU로 실행하기 import os os.en..
Programming/Tensorflow 2023. 1. 10. 10:12
이전 Prev 1 Next 다음