指定代码使用的GPU设备号

当你的主机上有多张GPU卡，可以通过如下方式设置代码main.py使用哪几张GPU卡来运行程序

import os 
os.environ['CUDA_VISIBLE_DEVICES'] = "0,1,2"  # 设定使用0,1,2号GPU卡

然后执行命令python main.py即可

注意，上述两行代码要作为main.py的第1、2行