环境变量设置GPU可见

python

os.environ['CUDA_VISIBLE_DEVICES']=str(index_and_partitions[0])

posted on 2017-09-21 12:36  reedlau  阅读(491)  评论(0)    收藏  举报

导航