首先 import os
然后通过 os.environ['CUDA_VISIBLE_DEVICES']=来设置用哪张卡
比如使用编号为0的卡:import os
os.environ['CUDA_VISIBLE_DEVICES']='0'
首先 import os
然后通过 os.environ['CUDA_VISIBLE_DEVICES']=来设置用哪张卡
比如使用编号为0的卡:import os
os.environ['CUDA_VISIBLE_DEVICES']='0'
转载于:https://www.cnblogs.com/yxxx/p/6888215.html