caffe （c++接口）多gpu操作函数及一些坑的记录

最新推荐文章于 2024-06-23 22:10:32 发布

jonaspku

最新推荐文章于 2024-06-23 22:10:32 发布

阅读量2k

点赞数

分类专栏： caffe 文章标签： caffe 多gpu gpu设置

本文链接：https://blog.csdn.net/Jonaspku/article/details/81067526

版权

本文记录了在Ubuntu 16.04系统中，使用Caffe进行多GPU操作的步骤，包括如何查看GPU状态、设置GPU模式以及避免GPU内存分配错误。在Caffe::set_mode(Caffe::GPU)后，需要使用Caffe::SetDevice(int gpuid)指定GPU设备，注意setdevice应在setmode之后，并且在多线程环境下要确保每个线程正确设置GPU ID，防止错误。ffmpeg的-loglevel error选项用于减少不必要的输出。

摘要由CSDN通过智能技术生成

记录一下。

软硬环境： ubuntu 16.04, Nvidia GTX 1080Ti *2

查看gpu（需要先安装驱动）：nvidia-smi

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.130                Driver Version: 384.130                   |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 108...  Off  | 00000000: