pytorch 指定GPU

 

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

 

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

导航