[Python] Tensorflow로 GPU할당하기
현재 imgaug를 통해 image augmentation을 진행중이다. https://github.com/aleju/imgaug GitHub - aleju/imgaug: Image augmentation for machine learning experiments. Image augmentation for machine learning experiments. - GitHub - aleju/imgaug: Image augmentation for machine learning experiments. github.com 그런데 그냥 CPU로만 데이터셋을 증강하려니까 너무 성능이 안좋았다.. 어제 퇴근 전 코드를 실행해놨는데 오늘 아침 출근하고보니 0.2%였다... 그래서 GPU를 이용하여 실행하기로 했다. (..
Python
2023. 6. 8. 11:39
