0

In the server,there are cudnn 7.0 and cuda 9.1 at /usr/local/cuda. I need to use detectron but it do not support cuda 9.1, so I need to install cuda 8.0.

I am using the server so I do not have root privilege.I tried to install cuda 8.0 and cudnn as this the second answer do at /home/xx/caffe2.

Then I build the caffe2, but it use the cuda at /usr/local/cuda again.

So is there any way to use the cuda at /home/xx/caffe2 ?

talonmies
  • 70,661
  • 34
  • 192
  • 269
qimeng wang
  • 83
  • 2
  • 6

1 Answers1

0

You need to specify environment variables when running the installation :

You can find a complete description of the environment variables in the setup.py file in the repository.

For example :

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

Ujjwal
  • 208
  • 1
  • 8