Когда вам нужная высокая скорость работы PyTorch, вы можете установить C++ библиотеку LibTorch. Этот тот же самый PyTorch только для C++. Особенно важна данная библиотека, когда вы развертываете свое приложение на сервере. При использовании библиотеки LibTorch могут возникнуть проблемы с включением omp.h. Данная проблема решается на операционных система Debian типа с помощью sudo apt-get install libomp-dev. Также необходимо вручную загрузить MNIST датасет. Обратите внимание на имена файлов MNIST, вместо точек нужно поставить тире.