CUDA相关

1、linux里查看gpu的显卡使用状态

  nvidia-smi

2、使用指定的gpu运行

  linux指令:CUDA_VISIBLE_DEVICES=7  xxx

  python文件:os.environ["CUDA_VISIBLE_DEVICES"] = "7"

posted @ 2022-06-10 11:00  一只小羊  阅读(36)  评论(0编辑  收藏  举报