pip install -r requirements.txt
python main.py
# CUDA_VISIBLE_DEVICES=2 python main.py # to specify GPU id to ex. 2Inspired by https://github.com/pytorch/examples/tree/master/mnist
| Name | Name | Last commit date | ||
|---|---|---|---|---|
pip install -r requirements.txt
python main.py
# CUDA_VISIBLE_DEVICES=2 python main.py # to specify GPU id to ex. 2Inspired by https://github.com/pytorch/examples/tree/master/mnist