PyTorch MNIST CPU 코딩

codingart (66)in #kr • 6 years ago (edited)

TensorFlow를 사용하는 MNIST 예제 즉 수기숫자 판독 문제는 머신러닝 학습의 핵심이라고 해도 지나치지 않다. PyTorch 사용법을 배워나가는데 있어서도 마찬가지라 할 수 있을 것이다. CPU 코딩이라 함은 GPU 가 없는 윈도우즈 10에 설치된 아나콘다에서 CPU 로 실행함을 뜻한다. GPU 로 실행하려면 cuda 코딩이 추가되어야 할 것이다.

한편 아나콘다에서 PyTorch 파이선 코드를 실행하기 위해서는 TensorFlow 와는 충돌이 있으므로 TensorFlow 가상환경과 별도로 Torch 가상환경을 만들어서 설치하여 코드르 실행시켜야 한다. 그렇지만 구글 Colabo를 사용하는 경우에는 그다지 문제가 되지 않는다. 구글 Colabo에서 GPU를 사용하여 실행하는 문제는 별도로 다루어 볼 계획이다.