Caffe CMake Pascal GPU Issue & Suggested Resolution

235 views
Skip to first unread message

CaffeStudent

unread,
Jun 21, 2016, 12:09:46 PM6/21/16
to Caffe Users
Hi,

I had a suggestion I wanted to run by the group for supporting the GTX 1080.

The use case I am running into an issue with is when I want to build caffe on a machine without a GPU, then install the built binaries on a machine with a GTX 1080 and run caffe.

If you take a look at

The default architecture list doesn't include "61", which is what the GTX 1080 reports
set(Caffe_known_gpu_archs "20 21(20) 30 35 50")

If Caffe is built on a machine which doesn't include the gpu, then "61" will not get auto detected and it will instead use the default architecture list, which does not include "61".  If this happens, the version of caffe that gets built will error out on a machine with a GTX 1080.

I would like to suggest adding "61" to the Caffe_known_gpu_archs list.  Is this a reasonable change to make?  It's probably easier for you guys to make the change, feel free to do so if what I am saying makes sense.

Developers will probably want to add a little more code to support CUDA_ARCH_NAME=Pascal as well.  I'm not really a gpu expert, so I'm not sure if it should add 60 and 61 or how exactly to handle the issue.

Thanks!

Kannan U V

unread,
Jul 28, 2016, 9:39:51 AM7/28/16
to Caffe Users
Did you get Caffe to run on GTX 1080?

I have issues compiling openCV on 1080, so built openCV without cuda. Then use openCV to build caffe, but errors out when running caffe.

I0728 09:31:13.549628 24404 layer_factory.hpp:76] Creating layer training_cells
F0728 09:31:13.676293 24404 layer_factory.hpp:68] Check failed: registry.count(type) == 0 (1 vs. 0) Layer type Convolution already registered.
*** Check failure stack trace: ***
    @     0x7f91dc68bdaa  (unknown)
    @     0x7f91dc68bce4  (unknown)
    @     0x7f91dc68b6e6  (unknown)
    @     0x7f91dc68e687  (unknown)
    @     0x7f91dc9d8ddb  (unknown)
Reply all
Reply to author
Forward
0 new messages