export CUDA_VISIBLE_DEVICES=0

python demo3.py