使用第二块显卡——CUDA——os.environ

仅作为记录,大佬请跳过

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1/'

你可能感兴趣的:(python,python)