如何使用指定的cuda和cudnn构建caffe2？

Question

在服务器中，/ usr / local / cuda中有cudnn 7.0和cuda 9.1。我需要使用Detectron，但它不支持CUDA 9.1，因此我需要安装CUDA 8.0。

我正在使用服务器，所以我没有root权限。我尝试在第二个答案/ home / xx / caffe2上以this的形式安装cuda 8.0和cudnn。

然后我构建了caffe2，但是它再次在/ usr / local / cuda中使用了cuda。

那么有什么方法可以在/ home / xx / caffe2使用cuda吗？

Answer 1

运行安装时需要指定环境变量：

您可以在存储库的setup.py文件中找到环境变量的完整说明。

例如：

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