技巧

2017-12-07 本文已影响0人梁新彦

指定gpu

    import os
    os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,4"