摘要:
对于数据在cpu与GPU之间同步的问题,caffe中用syncedMemory这个类来解 决;在GPU模式下,并且使用CUDA时,可以用CaffeMallocHost函数与CaffeFreeHost函数 来申请与释放内存; SyncedMemory的构造函数与析构函数不多说,可以看源码; 在该类中定义的变量有: void* cpu_ptr_; //数据在CPU上的指针; void* gpu... 阅读全文
摘要:
首先说明:Blob定义了一个类模板。 让我们看一下Blob的头文件里有什么哈: 定义了一个全局变量: const int kMaxBlobAxes = 32; 看看它的构造函数: Blob() : data_(), diff_(), count_(0), capacity_(0) {};explicit Blob(const int num, const int channels, const... 阅读全文