python指定gpu

import os

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

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

原文地址:https://www.cnblogs.com/Xycdada/p/13689704.html