训练如何选择GPU，设置GPU个数 

CUDA_VISIBLE_DEVICES=1 python run.py


or 

import os
os.environ["CUDA_VISIBLE_DEVICES"]="1"
posted @ 2017-02-28 17:33  Shiyu_Huang  阅读(2120)  评论(0)  编辑  收藏  举报