python指定gpu

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "1"　　

#  os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3" 

posted @ 2020-09-18 10:15  Xycdada  阅读(193)  评论(0)  编辑  收藏  举报