GPU

指定使用某个GPU

CUDA_VISIBLE_DEVICES=1 python xxx.py
CUDA_VISIBLE_DEVICES=0,1 python xxx.py

 

posted @ 2017-09-05 15:50  湖风雨晚晴  阅读(144)  评论(0)  编辑  收藏  举报