摘要: 服务器查看显卡 nvidia -smi # 实时查看 watch -n 0.5 nvidia-smi 指定GPU id import os os.environ['CUDA_VISIBLE_DEVICES'] = '1,0,2' import torch # 如果上述方法失效的话 torch.cud 阅读全文
posted @ 2022-04-15 10:27 嘎吱嘎吱脆 阅读(82) 评论(0) 推荐(0) 编辑