pytorch 指定GPU

 

os.environ['CUDA_VISIBLE_DEVICES']='3'

 

posted on 2020-01-09 10:30  兔子二  阅读(113)  评论(0)    收藏  举报

导航