2020年8月3日
深度学习中（tensorflow、pytorch）解决GPU使用冲突/抢卡
摘要： 问题：在运行深度学习模型的时候，总是要nvidia-smi一下看看那块显卡比较空闲，很麻烦。 解决方法：写个bash脚本，每次运行程序的时候，选择显存剩余最大的GPU。 #!/bin/bash n=$(nvidia-smi -q -d Memory|grep -A4 GPU|grep Free |g 阅读全文
posted @ 2020-08-03 09:37 蒙面的普罗米修斯 阅读(1156) 评论(0) 推荐(0) 编辑