Я бы хотел использовать caffe2 с поддержкой GPU.Я успешно установил caffe2 (Ubuntu 16.04, python2.7) со средой conda (команда: conda install pytorch-nightly -c pytorch)
Он успешно установлен (я проверил его с помощью команды: python2 -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure" и он говорит "Success")
Однако, когда я проверяю сборку caffe2 GPU (команда: python2 -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())), она возвращает 0.
У меня уже есть cuda, cuDNN, nccl, и я не понимаю, почему caffe2 не делаетобнаружить доступный графический процессор ..