Skip to content

qaohv/pytorch-performance-tuning-in-action

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch performance tuning in action

Code for article PyTorch performance tuning in action

Requirements

  1. NVIDIA driver == 455.32.00
  2. Docker >= 19.03
  3. nvidia-container-toolkit
  4. make

Dataset

Download dataset and put in data folder

Build docker image

docker build -t experiment .

Runnining

To run particular experiment use following command:

make expN

where N is number of experiment

To run all experiments use following command:

make all

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published