2021年9月11日
查看GPU运行情况(nvidia-smi )和指定GUP训练/推理模型（CUDA_VISIBLE_DEVICES使用）
摘要： 在深度学习模型训练或者推理的时候我们常常会需要使用到GPU加速，而在服务器上会存在多块GPU，并且可能会有很多人在使用这些GPU，因此如何设置程序在指定的GPU上训练模型/推理，下面介绍我们如何在python程序中指定GPU运行。 1、查看服务器显卡运行状况 在Linux服务器上我们可以使用以下命令 阅读全文
posted @ 2021-09-11 11:48 Circle_Wang 阅读(4786) 评论(0) 推荐(0) 编辑