jjldr / pytorch-cpp-1

C++ Implementation of PyTorch Tutorial for Deep Learning Researchers

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool


Build Status MIT License C++ PyTorch

This repository provides tutorial code in C++ for deep learning researchers to learn PyTorch.
Python Tutorial: https://github.com/yunjey/pytorch-tutorial

Getting Started

  • Fork/Clone and Install
$ git clone https://github.com/prabhuomkar/pytorch-cpp.git
$ chmod +x scripts.sh
$ ./scripts.sh install #optional: --cuda=(9.2 or 10.1) to install libtorch cuda versions (by default cpu version is installed) 
  • Download all datasets used in the tutorials
$ ./scripts.sh download_datasets
  • Build
$ ./scripts.sh build

Table of Contents

1. Basics

2. Intermediate

3. Advanced

Dependencies

  • C++
  • PyTorch C++ API

Authors

About

C++ Implementation of PyTorch Tutorial for Deep Learning Researchers

License:MIT License


Languages

Language:C++ 79.9%Language:CMake 14.0%Language:Shell 4.9%Language:Python 1.2%