data=data.cuda()
How to run a program on the GPU
CUDA_VISIBLE_DEVICES=O,1 python3 main.py //程序只能使用GPU:1