GPU
文章平均质量分 53
volcano_Lin
热爱CV和DL
展开
-
TensorFlow学习笔记[二]——在训练模型时如何指定具体哪一(多)块GPU显卡进行训练任务
在用tensorflow深度学习模型训练时,假设我们在训练之前没有指定具体用哪一块GPU进行训练,则默认的是选用第0块GPU来训练我们的模型。如果你的电脑有多块GPU的话,其它几块GPU的也会显示被占用。 有些时候,我们希望可以通过自己指定一块或者几块GPU来训练我们的模型,而不是用这种系统默认的方法。接下来将介绍三种指定GPU训练的方法。 我们...原创 2018-07-18 09:32:06 · 16531 阅读 · 8 评论 -
caffe深度学习【七】如何在Linux终端下关闭指定的GPU进程
首先,我们在终端输入nvidia-smi:nvidia-smi可以查看当前正常使用的GPU情况,如下图所示:在下面的Processes框框里,我们可以看到正常进行的进程,最左边一列是GPU的ID,第二列是我们待会需要用来终止进程的PID 当我们想要终止其中某个/或几个GPU进程,可通过以下命令实现:kill -9 PID(其中,PID就是刚刚我们通过nvi...原创 2018-08-08 14:21:40 · 5652 阅读 · 1 评论