arapaho.cpp

// ** NOTE: You would need to perform a "cudaSetDevice(gpuId)" with this same gpuId before any CUDA operations are performed on this gpuId

bool ArapahoV2::Setup(
            ArapahoV2Params & p,
            int & expectedWidth,
            int & expectedHeight
            int gpuId
                     )
{
    /// All old stuff
    load_weights(&net, p.weightfile);
    set_batch_network(&net, 1);     
    l = net.layers[net.n-1];
    
    // ** Add a new GPU index here instead of default
    net.gpu_index = gpuId;
}