Sou muito novo no Caffe, no CUDA, etc., mas consegui instalá-los e testá-los usando um exemplo da Caffe que tem como alvo a GPU e o Keras Example, que também se concentra na GPU. Então, tenho certeza que tudo está instalado. Eu tive que construir o CUDA para instalar, etc. Estou usando o Theano como back-end para o Keras.
Quando tento instalar os Dígitos usando o Pip3 (usando Python3) ou a instalação do APT, obtenho
The following packages have unmet dependencies. digits : Depends: python-caffe-nv (>= 0.13) but it is not going to be installed Depends: caffe-nv (>= 0.13) but it is not going to be installed Recommends: torch7-nv but it is not going to be installed E: Unable to correct problems, you have held broken packages.
Eu sei que o PyCaffe está funcionando? Desculpe se este é um problema newbie