0

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

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

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

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

4

1 回答 1

0

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

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

例如 ：

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

于 2018-09-11T18:38:42.913 回答