JittorCLS

A codebase for image classification based on Jittor

Getting Started

Install Jittor

sudo apt install python3.7-dev libomp-dev
python3.7 -m pip install jittor
python3.7 -m jittor.test.test_example
# If your computer contains an Nvidia graphics card, check the cudnn acceleration library
python3.7 -m jittor.test.test_cudnn_op

For more information on how to install jittor, you can check here.

Install OpenMPI

sudo apt install openmpi-bin openmpi-common libopenmpi-dev

To obtain more information about OpenMPI, you can check here.

Train

We provide scripts for single-machine single-gpu, single-machine multi-gpu training. Multi-gpu dependence can be referred to here

# Single GPU
bash train.sh

# Multiple GPUs
bash dist_train.sh

Test

# Single GPU
bash test.sh

# Multiple GPUs
bash dist_test.sh