Code for article PyTorch performance tuning in action
- NVIDIA driver == 455.32.00
- Docker >= 19.03
- nvidia-container-toolkit
- make
Download dataset and put in data folder
docker build -t experiment .
make expN
where N is number of experiment
make all