export CUDA_VISIBLE_DEVICES=0
python demo2.py