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!