Fork me on GitHub 0
摘要: os.environ["CUDA_VISIBLE_DEVICES"]="0" device=torch.device('cuda' if torch.cuda.is_available() else 'cpu') 阅读全文
posted @ 2020-08-07 17:30 amazingcode 阅读(517) 评论(0) 推荐(0) 编辑