plaidml1 Keras에서 GPU 사용하기 - MacBook 18' pro, Radeon pro 560X 4G 졸업 프로젝트를 위해 열심히 딥러닝을 공부하던 나. ConvNet 파트에 들어가자 모델 하나 학습 시키는데 1시간이 걸렸다. 슬픈 마음으로 Google Collab을 이용해야 하나 고민하고 있었는데, 내 친구가 맥북에 GPU가 있지 않냐고 했다. 그렇다. 나는 멀쩡한 GPU를 놔두고 CPU를 고문하고 있었던 것이다. 이렇게 좋은 노트북을 써본적이 없어서 나는 GPU가 들어있으리라고 생각도 못했었다 ㅋ큐ㅠㅠㅠㅠ 암튼 그래서 MacBook 18' pro, Radeon pro 560X 4G 환경에서 Keras GPU를 사용하는 법을 정리해봤다. 일단 tensorflow는 Nvidia 친화적이기 때문에 구글링하면 CUDA 세팅 어쩌구 밖에 안나온다. 좀더 찾아보니 plaidml과 Metal 을 이용한 방법이 있.. 2021. 2. 11. 이전 1 다음
