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


(Afshin Oroojlooy) #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