export CUDA_VISIBLE_DEVICES=0

python demo1.py