Skip to content

jiamny/pytorch_image_models

Repository files navigation

Train image models with pytorch c++

Train torch image models on the CIFAR10 data.

Prerequisites

  • cmake 3.18+
  • PyTorch 2.1.0
  • opencv 4.5.4/4.8.0
  • CUDA 11.8
  • cudnn 8.7
  • ubuntu 22.04
  • gcc-11/g++-11 for C++ 17

Links

About

Train image models with pytorch c++

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages