网站介绍：最近在测试实验室的服务器，试图跑了一下YoloV3，发现问题很多。其中一个就是：多块显卡被占用的问题，导致测试yoloV3时，程序无法运行。出现以上问题的原因是tensorflow在训练时默认占用所有GPU的显存。解决方法：在运行训练程序前，打开用户根目录下~/.bashrc文件在文件头配置环境：export CUDA_VISIBLE_DEVICES = NUMNUM是用户指定显卡的序号（0,1,2…），可以先用 nvidia-smi 查看当前哪块显卡可用。但这种方法限制了用户可见的GPU数量，_linux指定卡训练
- 链接地址：https://blog.csdn.net/woshiwangxin0/article/details/114536171
- 链接标题：Linux下使用GPU训练神经网络时 指定多块显卡的使用 解决tensorflow的占用问题_linux指定卡训练-CSDN博客
- 所属网站：blog.csdn.net
- 被收藏次数：7861
- 网站标签：linux指定卡训练
