Multi-gpu via torch::nn::parallel::data_parallel

C++
1

Hi,
I tried to use multi-gpu capability in C++, but when I write torch::nn::parallel::data_parallel, I am getting No member named 'parallel' in namespace 'torch::nn' error. Is torch::nn::parallel::data_parallel functional now?
Thanks,
Afshin

1 Like
2

you need
#include “torch/nn/parallel/data_parallel.h”