pycaffe support for multi GPU

[Guy N]

I saw that recently a new commit was inserted to Caffe, that should enable multi-GPU training with Python interface/Python custom layers.
I can't seem to train a network using multiple-GPU from my Python code.
I cloned the updated repository and built Caffe with USE_NCCL on.
The commands I tried to use in my Python script:

    caffe.set_mode_gpu()
    caffe.set_solver_count(2)

The error I get:

> F0223 10:27:41.249380 23468 python_layer.hpp:25] PythonLayer does not
> support CLI Multi-GPU, use train.py

When running the script from the command line, it simply uses 1 GPU only.

[Evan Shelhamer]

Use train.py: https://github.com/BVLC/caffe/blob/master/python/train.py

It handles the multi-GPU coordination for pycaffe.​

Evan Shelhamer

--
You received this message because you are subscribed to the Google Groups "Caffe Users" group.
To unsubscribe from this group and stop receiving emails from it, send an email to caffe-users+unsubscribe@googlegroups.com.
To post to this group, send email to caffe...@googlegroups.com.
To view this discussion on the web visit https://groups.google.com/d/msgid/caffe-users/54e949ee-defd-4925-b80c-875183aa89f4%40googlegroups.com.
For more options, visit https://groups.google.com/d/optout.