[pytorch] mnist

 

- mnist.py 소스 경로 : https://github.com/pytorch/examples/tree/master/mnist

 

1. 테스트 환경

- OS : Ubuntu 20.04.1 LTS
- Model : Apollo 6500 Gen10+ XL675d
- CPU : AMD EPYC 7543 32 Core 2.8GHz

 

2. 테스트 환경 구성 및 실행

$ conda create -n pytorch_cpu_1.9  $ source activate pytorch_cpu_1.9  $ conda install pytorch torchvision torchaudio cpuonly -c pytorch  $ export OMP_NUM_THREADS=1  $ python ./main.py

 

3. 결과

Threads 수	실행 시간(초)	비고
1	87.19
4	39.12
6	36.49
8	36.86
10	38.48
12	39.20
16	39.27
32	67.19