import os. import shutil. import time. import pprint. import torch. def set_gpu(x):. os.environ['CUDA_VISIBLE_DEVICES'] = x. print('using gpu:', x).
確定! 回上一頁