python指定GPU

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1" #指定第一块gpu

你可能感兴趣的:(python)