Caffe2 не может найти cuda-9.2 - PullRequest
Новая
       6

Caffe2 не может найти cuda-9.2

0 голосов
/ 26 ноября 2018

Я установил cuda-9.0 и cuda-9.2.Я обычно использую cuda-9.0, но для установки caffe2 мне нужно использовать cuda-9.2.Я настраиваю файл cuda9.2-env, который содержит эти строки: 

export PATH=$PATH:/usr/local/cuda-9.2/bin
export CUDADIR=/usr/local/cuda-9.2
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.2/lib64

после активации моего conda env, я делаю:

source cuda9.2-env

Мой файл .bashrc содержит: 

export LD_LIBRARY_PATH=/usr/local/cuda/lib64/
export PATH=/usr/local/cuda/

Но сценарий установки caffe2 по-прежнему обнаруживает cuda-9.0 (/ usr / local / cuda) !!

...