Fork me on GitHub 0
2020年8月7日
指定GPU显卡
摘要： os.environ["CUDA_VISIBLE_DEVICES"]="0" device=torch.device('cuda' if torch.cuda.is_available() else 'cpu') 阅读全文
posted @ 2020-08-07 17:30 amazingcode 阅读(512) 评论(0) 推荐(0) 编辑