Как собрать caffe2 с указанными cuda и cudnn?

Question

На сервере есть cudnn 7.0 и cuda 9.1 в / usr / local / cuda. Мне нужно использовать детектор, но он не поддерживает cuda 9.1, поэтому мне нужно установить cuda 8.0.

Я использую сервер, поэтому у меня нет привилегий суперпользователя. Я попытался установить cuda 8.0 и cudnn, так как это второй ответ сделать в /home/xx/caffe2.

Затем я создаю caffe2, но он снова использует cuda в / usr / local / cuda.

Так есть ли способ использовать cuda в /home/xx/caffe2?

Answer 1

Вам нужно указать переменные среды при запуске установки:

Вы можете найти полное описание переменных среды в setup.py файл в хранилище.

Например:

CUDA_HOME=/home/xxx/cuda-9.1 python setup.py install