Ubuntu指定GPU执行程序

法1：

CUDA_VISIBLE_DEVICES=0,1,2 python test.py

法1中，0,1,2为显卡的编号。可以选择多块或单个显卡

法2：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2"

法2将GUP的指定写在程序中。